From a1c7fe1b6d95e2560e62c12453da287d36d4a714 Mon Sep 17 00:00:00 2001 From: Charles Cabergs Date: Sat, 29 Aug 2020 17:13:57 +0200 Subject: Removed bloat, Added logreg_train --- src/scatter_plot.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) mode change 100644 => 100755 src/scatter_plot.py (limited to 'src/scatter_plot.py') diff --git a/src/scatter_plot.py b/src/scatter_plot.py old mode 100644 new mode 100755 index 74e0384..cbb38b9 --- 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() -- cgit