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_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)
 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)