1 files changed, 14 insertions, 3 deletions
diff --git a/src/pair_plot.py b/src/pair_plot.py
index bf0c632..37821eb 100644..100755
--- a/src/pair_plot.py
+++ b/src/pair_plot.py
@@ -1,6 +1,17 @@
-from analysis import Analysis
+#!/bin/python3
+
+import sys
+
+import pandas as pd
+import matplotlib.pyplot as plt
+
+from dataset import Dataset
 
 
 if __name__ == '__main__':
-    a = Analysis('../datasets/dataset_train.csv')
-    a.pair_plot()
+    if len(sys.argv) != 2:
+        raise "Usage: {} dataset_path".format(sys.argv[0])
+    d = Dataset(sys.argv[1])
+    pd.plotting.scatter_matrix(d.df_scores, s=2, alpha=0.8)
+    plt.show()
+