Add new target

gatewayd-io · Oct 27, 2024 · d0bfbd5 · d0bfbd5
1 parent 8ff5ca1
commit d0bfbd5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/training/Makefile b/training/Makefile
@@ -11,3 +11,10 @@ train-dataset2:
 	# Add warm-up to the model to load the model into memory after
 	# starting the Serving API container
 	python warm_up.py ../dataset/sqli_dataset2.csv ../sqli_model/2
+
+train-dataset2-model3:
+	# Train the model with dataset2
+	poetry run python train_v3.py ../dataset/sqli_dataset2.csv ../sqli_model/3
+	# Add warm-up to the model to load the model into memory after
+	# starting the Serving API container
+	poetry run python warm_up.py ../dataset/sqli_dataset2.csv ../sqli_model/3