if 0:
model.save("neuralNetModel.pkl")
#%% Model creation
if 1:
model = NeuralNet(Epochs=1e4,
learning_rate=1e-6,
batch_size=10000,
verbose=False,
regularisation=1e-9)
model.add_layer(nnet_layer(2304, 200))
model.add_layer(nnet_layer(200, 7))
refit = False
else:
model.Epochs = 1e4
model.learning_rate = 1e-6
model.batch_size = 10000
model.verbose = False
model.regularisation = 1e-9
# model.Epochs = 100000
# model.learning_rate = 1e-6
#
#%% Training
if 1:
model.train(X_train, T_train)
#%% Predicting and testing
if 1:
