|
|
--- |
|
|
license: mit |
|
|
tags: |
|
|
- sentiment_analysis |
|
|
- mental_health_disorder |
|
|
- text-classification |
|
|
- pytorch |
|
|
- chatbot |
|
|
- transformers |
|
|
--- |
|
|
# 🧠 Mental Health Chatbot |
|
|
|
|
|
This repository contains an AI-powered chatbot designed to assist users by analyzing their text input for **sentiment** and **mental health disorders**, and providing personalized recommendations based on AI predictions and explanations. |
|
|
|
|
|
--- |
|
|
|
|
|
## ✨ Features |
|
|
|
|
|
- **💬 Sentiment Analysis**: Classifies user input as Positive or Negative. |
|
|
- **🧠 Disorder Classification**: Predicts potential mental health conditions like Anxiety, Depression, etc. |
|
|
- **🚦 Risk Level Estimation**: Assesses the severity of user input (Low, Medium, High). |
|
|
- **🧭 Personalized Recommendations**: Suggests helpful actions based on predictions. |
|
|
- **🧪 LIME Interpretability**: Explains model predictions using LIME. |
|
|
- **🖥️ Terminal Chat Interface**: Human-like, emoji-enhanced chatbot interaction in the terminal. |
|
|
|
|
|
--- |
|
|
|
|
|
## ✅ Requirements |
|
|
|
|
|
- Python 3.7+ |
|
|
- PyTorch |
|
|
- Hugging Face Transformers |
|
|
- LIME |
|
|
- Scikit-learn |
|
|
|
|
|
--- |
|
|
|
|
|
## 📦 Installation |
|
|
|
|
|
1. **Clone the repository:** |
|
|
|
|
|
```bash |
|
|
git clone https://huggingface.co/username/MentalHealthChatbot |
|
|
cd MentalHealthChatbot |
