Skip to content

A hands-on ML project to classify movie reviews as positive or negative using a simple and handwritten TF-IDF + Logistic Regression baseline.

Notifications You must be signed in to change notification settings

staticvoidmainmaui/Sentiment-Analysis-IMDB

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

# Sentiment Analysis on IMDB (Work in Progress)

A hands-on ML project to classify movie reviews as positive or negative using a simple and handwritten TF-IDF + Logistic Regression baseline.

This repo is a starting point and something personalized for AI Projects at UF.

IMDB Reviews received from acquaintance, John Burrow.

Guidelines coming from inspiration of other repositories.


## Quickstart

python -m venv .venv

\# Windows

.\\.venv\\Scripts\\Activate

\# macOS/Linux

\# source .venv/bin/activate

pip install -r requirements.txt

python train.py



---




## Status

Work in progress; baseline runs and prints accuracy + demo predictions.

**Sample Output (tiny sample space):**

Accuracy: 0.333

  precision recall f1-score support

  0 0.000 0.000 0.000 2

  1 0.333 1.000 0.500 1

  accuracy 0.333 3

  macro avg 0.167 0.500 0.250 3

weighted avg 0.111 0.333 0.167 3

'This was amazing, I loved every minute.' -> positive

'The worst film I have ever seen.' -> positive

## Roadmap

- Expand dataset and preprocessing (full IMDB dataset)

About

A hands-on ML project to classify movie reviews as positive or negative using a simple and handwritten TF-IDF + Logistic Regression baseline.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages