1 files changed, 14 insertions, 3 deletions
diff --git a/src/scatter_plot.py b/src/scatter_plot.py
index 74e0384..cbb38b9 100644..100755
--- a/src/scatter_plot.py
+++ b/src/scatter_plot.py
@@ -1,6 +1,17 @@
-from analysis import Analysis
+#!/bin/python3
+
+import sys
+
+import matplotlib.pyplot as plt
+
+from dataset import Dataset
 
 
 if __name__ == '__main__':
-    a = Analysis('../datasets/dataset_train.csv')
-    a.scatter()
+    if len(sys.argv) != 2:
+        raise "Usage: {} dataset_path".format(sys.argv[0])
+    d = Dataset(sys.argv[1])
+    plt.scatter(d.df['astronomy'], d.df['defense_dark_arts'], s=5)
+    plt.xlabel('astronomy')
+    plt.ylabel('defense_dark_arts')
+    plt.show()