diff --git a/src/models/overview_nn.py b/src/models/overview_nn.py index a4b054c..fe6d4bb 100644 --- a/src/models/overview_nn.py +++ b/src/models/overview_nn.py @@ -51,7 +51,7 @@ import matplotlib.pyplot as plt # Plot training & validation accuracy values -plt.plot(history.history['acc']) +plt.plot(history.history['accuracy']) plt.title('Model accuracy') plt.ylabel('Accuracy') plt.xlabel('Epoch') @@ -109,4 +109,4 @@ import json with open('dominostats.json', 'w') as f: f.write(json.dumps({"Precision": prec_mean, "Recall": rec_mean})) -print("Saved metrics to dominostats.json. You should be able to see these on the Jobs Dashboard.") \ No newline at end of file +print("Saved metrics to dominostats.json. You should be able to see these on the Jobs Dashboard.")