Skip to content

A Natural Language Prediction Model which will take written words (string) as input and classify it into the three categories, i.e. hate, neutral, supportive.

Notifications You must be signed in to change notification settings

chinmay-1302/sentiment_pred

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Hate Speech Detection Model

HSD-Github-2

This repo is to maintain the web app for the hate speech detection model. The model is trained on the HASOC dataset and is deployed using FastAPI. The model is trained using the pretrained distilBERT model and the training code is available in the model folder.

💬 Problem Statement

A Natural Language Prediction Model which will take written words (string) as input and classify it into the three categories, i.e. hate, neutral, supportive.

💻 Tech Stack

  • Python
  • Pandas
  • Numpy
  • Scikit-learn
  • Matplotlib
  • PyTorch
  • Transformers
  • DistilBERT
  • FastAPI
  • React

🧠 Model

  • DistilBERT

⚙️ Working

  • The model is trained on the HASOC dataset.
  • The model is trained using the pretrained distilBERT model.
  • The model is deployed using FastAPI.
  • The web app is built using React.

📌 Setup

  • Clone the repository.
  • Install the required packages using pip install -r requirements.txt.
  • Run the FastAPI server using uvicorn main:app --reload.
  • Run the React app using npm start.

📚 References

About

A Natural Language Prediction Model which will take written words (string) as input and classify it into the three categories, i.e. hate, neutral, supportive.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published