Loading…
How it works
- Academic data → Label encoding, SMOTE, StandardScaler → XGBoost → Depression
- Workplace survey → Label encoding, SMOTE, StandardScaler → Random Forest + GridSearchCV → Treatment
- Social media text → TF-IDF (10k features) → Linear SVM → Severe distress
Models were trained in Google Colab, serialized with joblib and are served by a Flask API.