def test_train_nn_with_fclayers(self): X_train, y_train, X_test, y_test = generate_data(1000, 10) y_model = graph.train_nn(X_train, y_train, X_test, [10, 22, 37], 50, 2, 0.001)
def test_train_nn(self): X_train, y_train, X_test, y_test = generate_data(1000, 10) y_model = graph.train_nn(X_train, y_train, X_test, [], 50, 2, 0.001)