add epochs

lukas · lukas · commit 1df2e91a5e9c · 2018-04-09T07:51:33.000-07:00
diff --git a/keras-perceptron/perceptron-linear.py b/keras-perceptron/perceptron-linear.py
@@ -30,7 +30,7 @@
                 metrics=['accuracy'])
 
 # Fit the model
-model.fit(X_train, y_train, validation_data=(X_test, y_test),
+model.fit(X_train, y_train, epochs=10, validation_data=(X_test, y_test),
                     callbacks=[WandbKerasCallback()])
 
 
