From 082f1bfc4e920a030eaadcf050dd1816324a222d Mon Sep 17 00:00:00 2001 From: ddl-aambekar <55107387+ddl-aambekar@users.noreply.github.com> Date: Wed, 8 Jul 2020 17:34:50 -0700 Subject: [PATCH] Update overview_nn.py adding fix for plt.plot(history.history['acc']) --> KeyError: 'acc' --- src/models/overview_nn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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.")