https://github.com/web3batman/pytorch-sentiment-analysis
https://github.com/web3batman/pytorch-sentiment-analysis
Last synced: about 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/web3batman/pytorch-sentiment-analysis
- Owner: web3batman
- Created: 2024-05-17T17:41:48.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-05-17T17:43:02.000Z (about 2 years ago)
- Last Synced: 2025-01-30T05:15:28.495Z (over 1 year ago)
- Language: Python
- Size: 1.95 KB
- Stars: 3
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Sentiment Analysis with PyTorch and Hugging Face IMDB Dataset
This project implements a sentiment analysis model using PyTorch and the IMDB dataset from the Hugging Face datasets library. The model uses an LSTM architecture to classify movie reviews as either positive or negative.
## Features
- Loads the IMDB dataset from the Hugging Face datasets library
- Tokenizes the text data using the BERT tokenizer
- Trains a sentiment analysis model using PyTorch and LSTM
- Evaluates the model's performance on the test set
- Provides a Flask API for sentiment prediction
## Requirements
- Python 3.7 or higher
- PyTorch
- Hugging Face Transformers
- Hugging Face Datasets
- Flask
- Flask-CORS