Graph make rule

Author: devExcale

makefile

@@ -139,6 +139,10 @@ report-check:
 		fi ; \
 	done ; \
 
+graph:
+	@echo "Generating graph for $(dataset)"
+	@python src/graph.py -d $(dataset) csv csv
+
 extract-medpub:
 	@echo "--- Exporting medpub dataset ---"
 	@cd .datasets && python extract_csv.py --offset 1 --threshold 600 --column "abstract" train.csv medpub

src/graph.py

@@ -1,19 +1,13 @@
 import argparse
+from typing import Dict
 
 import matplotlib.pyplot as plt
 import pandas
-from typing import Dict
 
 parser = argparse.ArgumentParser(
 	description="Generate a graph with the execution times of the MPI/OMP MinHash implementations."
 )
 
-parser.add_argument(
-	"-p", "--processes",
-	required=True,
-	type=int,
-	help="Max number of processes in the csv file",
-)
 parser.add_argument(
 	"-d", "--dataset",
 	type=str,
@@ -36,7 +30,6 @@
 
 def draw_graph(
 		dists: Dict[str, pandas.DataFrame],
-
 		x_label: str,
 		y_label: str,
 		title: str = None,
@@ -63,7 +56,6 @@ def draw_graph(
 	for name, df in dists.items():
 		plt.plot(df[x_label], df[y_label], label=name, marker='o', markersize=3)
 
-
 	# Add labels
 	plt.title(title)
 	plt.xlabel(x_label)
@@ -115,7 +107,7 @@ def get_dataframe(csv_path: str) -> pandas.DataFrame:
 
 	# Read the CSV files
 	csv_paths = {
-		lib: f"{args.in_csv_path}/time_{lib}_{args.dataset}_{args.processes}.csv"
+		lib: f"{args.in_csv_path}/time_{lib}_{args.dataset}.csv"
 		for lib in ["MPI", "OMP"]
 	}
 	data = {
@@ -124,7 +116,7 @@ def get_dataframe(csv_path: str) -> pandas.DataFrame:
 	}
 
 	# Compute save path
-	save_path = f"{args.out_png_path}/time_{args.dataset}_{args.processes}.png" if args.out_png_path else None
+	save_path = f"{args.out_png_path}/time_{args.dataset}.svg" if args.out_png_path else None
 
 	# Draw a graph
 	draw_graph(