imdb-lstm-sentiment-model

IMDB LSTM Sentiment Analysis Model

This model predicts the sentiment of movie reviews as either positive or negative using an LSTM architecture.

Model Description

• Model Type: LSTM
• Input: Movie review text
• Output: Sentiment label (positive/negative)

Usage

To use this model, you can call the predict_sentiment(review_text) function in the provided inference.py script.

Example

from transformers import pipeline

# Load the sentiment-analysis pipeline
model_name = "your-username/imdb-lstm-sentiment-model"  # Replace with your model's name
sentiment_analyzer = pipeline("text-classification", model=model_name)

# Test the model with a review
review = "This movie was fantastic!"
result = sentiment_analyzer(review)
print(result)
'''