diff --git a/training/Makefile b/training/Makefile index cf81602..1a418b9 100644 --- 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