Skip to content

This project classifies Reddit user posts as either "Depressed" or "Not Depressed" using Sentence-BERT embeddings and a convolutional neural network. It includes an end-to-end NLP pipeline with preprocessing, model training, evaluation, and real-time text input prediction.

License

Notifications You must be signed in to change notification settings

avdvh/DeepPressNet

Repository files navigation

DeepPressNet: Depression Detection from Text using SBERT + Neural Networks

DeepPressNet is a deep learning pipeline that classifies Reddit posts as either "Depressed" or "Not Depressed" based on textual input. It leverages Sentence-BERT (SBERT) for embedding generation and a Multilayer Perceptron (MLP) neural network optimized using Optuna for classification.

Try It Live: Click Here

Features

  • End-to-end NLP pipeline
  • SBERT for powerful semantic embeddings
  • Optuna-based hyperparameter tuning
  • Evaluation with precision, recall, F1, ROC AUC, confusion matrix
  • CSV logs and .txt reports for model training
  • Real-time sentence prediction with confidence score
  • Visualizations: training curves, metric comparisons

Tech Stack

  • SBERT (sentence-transformers)
  • TensorFlow / Keras
  • Optuna for hyperparameter optimization
  • Pandas, NumPy, Scikit-learn, Seaborn, Matplotlib

Model Performance

image image image

After Optuna Fine-Tuning

image image image

Real-Time Prediction

image

About

This project classifies Reddit user posts as either "Depressed" or "Not Depressed" using Sentence-BERT embeddings and a convolutional neural network. It includes an end-to-end NLP pipeline with preprocessing, model training, evaluation, and real-time text input prediction.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published