Fixed mlp

Author: lukas

videos/intro/mlp.py

@@ -12,6 +12,8 @@
 run = wandb.init()
 config = run.config
 
+config.epochs = 10
+config.optimizer = "adam"
 config.hidden_nodes = 100
 
 # load data
@@ -43,4 +45,4 @@
 # Fit the model
 model.fit(X_train, y_train, validation_data=(X_test, y_test), 
       epochs=config.epochs,
-      callbacks=[WandbCallback(validation_data=X_test, labels=labels)])
+      callbacks=[WandbCallback(data_type="image", labels=labels)])