def DigitRecognitionTest():
X_train, y_train, X_val, y_val, X_test, y_test = load_and_extract_mnist_data.load_dataset()
X_train = X_train.reshape(np.size(X_train,0),28*28).transpose()
X_val = X_val.reshape(np.size(X_val,0),28*28).transpose()
X_test = X_test.reshape(np.size(X_test,0),28*28).transpose()
Y_train = []
for e in y_train:
tmp = [0]*10
tmp[e]=1
Y_train.append(tmp)
Y_train = np.array(Y_train).transpose()
Y_val = []
for e in y_val:
tmp = [0]*10
tmp[e]=1
Y_val.append(tmp)
Y_val = np.array(Y_val).transpose()
Y_test = []
for e in y_test:
tmp = [0]*10
tmp[e]=1
Y_test.append(tmp)
Y_test = np.array(Y_test).transpose()
myann = NeuralNetwork('Digital Recognition')
myann.model.SetInputLayer(28*28)
myann.model.SetOutputLayer('SoftMax')
myann.model.SetFullConnectLayer([(50,'ReLu'),(20,'ReLu'),(20,'ReLu'),(10,None)])
myann.setTrain(X_train,Y_train)
myann.setCVD(X_val,Y_val)
myann.setTest(X_test,Y_test)
myann.MiniBatch_Train(1000,2500,True)
