ASledziewska commited on
Commit
3d6f897
1 Parent(s): 2cc61e5

Update xgb_mental_health.py

Browse files
Files changed (1) hide show
  1. xgb_mental_health.py +1 -1
xgb_mental_health.py CHANGED
@@ -68,7 +68,7 @@ if __name__ == "__main__":
68
  y_test, y_pred = mental_classifier.train_model(X, y)
69
  mental_classifier.save_model()
70
  else:
71
- mental_classifier.load_model()
72
  mental_classifier.initialize_tokenizer(tokenizer_model_name) # Ensure tokenizer is initialized if loading model from pickle
73
  mental_classifier.preprocess_data()
74
 
 
68
  y_test, y_pred = mental_classifier.train_model(X, y)
69
  mental_classifier.save_model()
70
  else:
71
+ mental_classifier.load_model(model_path)
72
  mental_classifier.initialize_tokenizer(tokenizer_model_name) # Ensure tokenizer is initialized if loading model from pickle
73
  mental_classifier.preprocess_data()
74