# -*- coding: utf-8 -*-
"""
Created on Wed Nov 13 12:49:20 2019

@author: Ashima
"""


import os

#Directory Information
ROOT_DIR = os.path.abspath('../')
DATA_DIR = os.path.join(ROOT_DIR, "DATASET/")
OUT_DIR = os.path.join(ROOT_DIR, 'RESULT/')

TRAIN_INPUT = "train-images-idx3-ubyte.gz"
TRAIN_LABELS = "train-labels-idx1-ubyte.gz"

TEST_INPUT = "t10k-images-idx3-ubyte.gz"
TEST_LABELS = "t10k-labels-idx1-ubyte.gz"

MNIST_FILE = "MNIST_Subset.h5"

WEIGHTS_FILE = "weights"

#Dataset Information
NUM_SAMPLES = 14251
#NUM_TEST = 10000
IMAGE_SIZE = 28
CLASSES = 10
CLASS_LABELS = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
SPLIT_FACTOR = 15
NUM_VAL = int(round((SPLIT_FACTOR/100.)*NUM_SAMPLES))
NUM_TEST = NUM_VAL
NUM_TRAIN = int(NUM_SAMPLES - (2*NUM_VAL))
#Model Parameters
LEARNING_RATE = 0.2
NUM_EPOCHS = 100

SEED = 128
