Fix OSError in NLTK check and use HistGradientBoostingClassifier

models/classifier.py

@@ -10,7 +10,7 @@ import joblib
 import os
 from typing import Tuple, Dict, List, Optional
 from sklearn.linear_model import LogisticRegression
-from sklearn.ensemble import GradientBoostingClassifier, VotingClassifier
+from sklearn.ensemble import HistGradientBoostingClassifier, VotingClassifier
 from sklearn.model_selection import train_test_split, cross_val_score
 from sklearn.metrics import (
     classification_report,
@@ -118,8 +118,8 @@ class DistressClassifier:
         lr = LogisticRegression(
             max_iter=1000, C=1.0, class_weight="balanced", random_state=42
         )
-        gb = GradientBoostingClassifier(
-            n_estimators=200, max_depth=5, learning_rate=0.1, random_state=42
+        gb = HistGradientBoostingClassifier(
+            max_iter=100, max_depth=5, learning_rate=0.1, random_state=42
         )
 
         self.classifier = VotingClassifier(