holiday similarity update

1 files changed, 19 insertions, 0 deletions

diff --git a/cv/holiday_similarity/vis_utils.py b/cv/holiday_similarity/vis_utils.py
new file mode 100644
index 0000000..0910364
--- /dev/null
+++ b/cv/holiday_similarity/vis_utils.py
@@ -0,0 +1,19 @@
+import matplotlib.pyplot as plt
+
+
+def plot_training_curve(history):
+    # print(history.history.keys())
+    plt.subplot(211)
+    plt.title("Loss")
+    plt.plot(history.history["loss"], color="r", label="train")
+    plt.plot(history.history["val_loss"], color="b", label="validation")
+    plt.legend(loc="best")
+
+    plt.subplot(212)
+    plt.title("Accuracy")
+    plt.plot(history.history["accuracy"], color="r", label="train")
+    plt.plot(history.history["val_accuracy"], color="b", label="validation")
+    plt.legend(loc="best")
+
+    plt.tight_layout()
+    plt.show()