s

train.py

@@ -57,7 +57,7 @@ def train_model(data_file, model_file, epochs, batch_size, test_size, random_sta
     model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
 
     model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size)
-    loss, accuracy = model.evaluate(X_test, y_test)
+    accuracy = model.evaluate(X_test, y_test)
     print('Test accuracy:', accuracy)
 
     model.save(model_file)
@@ -69,3 +69,5 @@ def run_experiment():
     accuracy = train_model()
     ex.log_scalar('accuracy', accuracy)  
     ex.add_artifact('model.h5')  
+
+ex.run()