Edit model card

Model Card for SentimentTensor

This modelcard provides details about the SentimentTensor model, developed by Saish Shinde, for sentiment analysis using LSTM architecture.

Model Details

Model Description

The SentimentTensor model is a deep learning model based on LSTM architecture, developed by Saish Shinde, for sentiment analysis tasks. It achieves an accuracy of 81% on standard evaluation datasets. The model is designed to classify text data into three categories: negative, neutral, and positive sentiments.

  • Developed by: Saish Shinde
  • Model type: LSTM-based Sequence Classification
  • Language(s) (NLP): English
  • License: No specific license

Dataset Used

yelp dataset with 4.04GB compressed,8.65GB uncompressed data

Uses

Direct Use

The SentimentTensor model can be directly used for sentiment analysis tasks without fine-tuning.

Downstream Use

This model can be fine-tuned for specific domains or integrated into larger NLP applications.

Out-of-Scope Use

The model may not perform well on highly specialized or domain-specific text data.

Bias, Risks, and Limitations

The SentimentTensor model, like any LSTM-based model, may have biases and limitations inherent in its training data and architecture. It might sometimes struggle with capturing long-range dependencies or understanding context in complex sentences, also it emphasizes less on neutral sentiment

Recommendations

Users should be aware of potential biases and limitations and evaluate results accordingly.

How to Get Started with the Model

Loading the Model

You can load the SentimentTensor model using the Hugging Face library:

python Code:

from transformers import AutoModelForSequenceClassification, AutoTokenizer

Load the model and tokenizer

model = AutoModelForSequenceClassification.from_pretrained("your-model-name") tokenizer = AutoTokenizer.from_pretrained("your-tokenizer-name")

Tokenization

text = "Your text data here" tokenized_input = tokenizer(text, return_tensors="pt")

Sentiment Analysis

#Forward pass through the model outputs = model(**tokenized_input)

#Get predicted sentiment label predicted_label = outputs.logits.argmax().item()

Example Usage


from transformers import AutoModelForSequenceClassification, AutoTokenizer

# Load the model and tokenizer
model = AutoModelForSequenceClassification.from_pretrained("saishshinde15/SentimentTensor")
tokenizer = AutoTokenizer.from_pretrained("saishshinde15/SentimentTensor")

# Tokenize text data
text = "This is a great movie!"
tokenized_input = tokenizer(text, return_tensors="pt")

# Perform sentiment analysis
outputs = model(**tokenized_input)
predicted_label = outputs.logits.argmax().item()

# Print predicted sentiment
sentiment_labels = ["negative", "neutral", "positive"]
print(f"Predicted Sentiment: {sentiment_labels[predicted_label]}")

Model Architecture and Objective

The SentimentTensor model is based on LSTM architecture, which is well-suited for sequence classification tasks like sentiment analysis. It uses long short-term memory cells to capture dependencies in sequential data.

Model Card Authors

Saish Shinde

Downloads last month
11
Inference Examples
Inference API (serverless) does not yet support keras models for this pipeline type.

Dataset used to train saishshinde15/SentimentTensor