add prints

tasks/text.py

@@ -33,6 +33,7 @@ def baseline_model(dataset_length: int):
     return predictions
 
 def tree_classifier(test_dataset: dict, model: str):
+    print("Starting tree model run")
 
     texts = test_dataset["quote"]
 
@@ -47,6 +48,8 @@ def tree_classifier(test_dataset: dict, model: str):
                      'xgboost.sklearn.XGBClassifier'])
 
     predictions = model.predict(texts)
+    print("Finished tree model run")
+
 
     return predictions
 
@@ -71,6 +74,7 @@ class TextDataset(Dataset):
 
 
 def bert_classifier(test_dataset: dict, model: str):
+    print("Starting BERT model run")
     texts = test_dataset["quote"]
 
     model_repo = f"theterryzhang/frugal_ai_{model}"
@@ -97,6 +101,8 @@ def bert_classifier(test_dataset: dict, model: str):
             p = torch.argmax(outputs.logits, dim=1)
             predictions = np.append(predictions, p.cpu().numpy())
     
+    print("Finished BERT model run")
+    
     return predictions