The Suicidal Ideation Detection project is designed to identify and assess suicidal thoughts from social media posts. Utilizing advanced natural language processing (NLP) techniques, this project leverages a fine-tuned LSTM model based on BERT from TensorFlow Hub to analyze text data. The model achieves an impressive 97% accuracy and performs well on false negatives, ensuring reliable detection.
Suicidal ideation is a critical issue that can be identified through social media activity. This project aims to create a safer online environment by detecting signs of suicidal thoughts in social media posts. Data is collected from Reddit, Twitter, and Tumblr using their respective APIs and is annotated by our team. The backend functionality is showcased using Django, with potential to expand the system as an API or integrate it into a Chrome extension. This model could also be adopted by social media platforms to enhance user safety.
- High Accuracy Detection: Leveraging an LSTM-based model fine-tuned with BERT for precise suicidal ideation detection.
- Data Collection: Utilizes Reddit, Twitter, and Tumblr APIs for comprehensive social media data.
- Backend Implementation: Showcased through a Django application, with potential to expose as an API or integrate into other platforms.
- Future Integration: Possibility to develop a Chrome extension or integrate into social media platforms for enhanced safety.
- Clone the Repository:
git clone https://github.com/yourusername/suicidal-ideation-detection.git
cd suicidal-ideation-detection
- Create and Activate a Virtual Environment:
python -m venv venv
source venv/bin/activate # On Windows use `venv\Scripts\activate`
- Install Dependencies:
pip install -r requirements.txt
- Run Migrations:
python manage.py migrate
- Start the Django Development Server:
python manage.py runserver
- Access the Application: Open your web browser and navigate to Localhost to view the Django application.
- Backend Functionality: The model processes social media posts to identify potential suicidal ideation.
- API Exposure: Future developments may include exposing the functionality as an API for integration.
- Chrome Extension: Plans to create a Chrome extension for real-time analysis of social media posts.
Contributions to this project are welcome. Please fork the repository and submit a pull request with your changes.
This project is licensed under the Apache-2.0 License. See the LICENSE file for details.
Prachit Mhalgi |
Riddhi Doshi |
Athmika Hebbar |
Aditi Bhagwat |