Spaces:
Sleeping
Sleeping
ASledziewska
commited on
Commit
•
8c1f1b5
1
Parent(s):
f52be0e
Update xgb_mental_health.py
Browse files- 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()
|
72 |
mental_classifier.initialize_tokenizer(tokenizer_model_name) # Ensure tokenizer is initialized if loading model from pickle
|
73 |
mental_classifier.preprocess_data()
|
74 |
|