def test(train_X, test_X, train_y, test_y, args):
filename = FILENAME_USED_NN
if len(args) == 3:
filename = "trainedModels/" + args[2]
model_loaded = NeuralNet.load(filename)
model = NeuralNet(model_loaded)
print("final train accuracy:", model.score(train_X, train_y))
print("final test accuracy:", model.score(test_X, test_y))
