updated path
Browse files- xgb_mental_health.py +2 -2
xgb_mental_health.py
CHANGED
@@ -67,8 +67,8 @@ class MentalHealthClassifier:
|
|
67 |
|
68 |
if __name__ == "__main__":
|
69 |
tokenizer_model_name = "nlptown/bert-base-multilingual-uncased-sentiment"
|
70 |
-
data_path = 'data/
|
71 |
-
model_path = '
|
72 |
mental_classifier = MentalHealthClassifier(data_path, model_path)
|
73 |
|
74 |
if not os.path.exists(model_path):
|
|
|
67 |
|
68 |
if __name__ == "__main__":
|
69 |
tokenizer_model_name = "nlptown/bert-base-multilingual-uncased-sentiment"
|
70 |
+
data_path = 'data/data.csv'
|
71 |
+
model_path = 'mental_health_model.pkl'
|
72 |
mental_classifier = MentalHealthClassifier(data_path, model_path)
|
73 |
|
74 |
if not os.path.exists(model_path):
|