XXuSea/my-pytorch-model-LSTM
By XXuSea · ↓ 0 · ♥ 0 · Updated 2025-03-05
What's in this repo
This page lists the files in XXuSea/my-pytorch-model-LSTM, including imdb_sentiment_model.onnx.
- 1 ONNX models (.onnx files)
- 5 files, including 1 .md, 1 .json, 1 .onnx, 1 .pkl
- Last updated 2025-03-05
ONNX files
| File | Size | Download |
|---|---|---|
| imdb_sentiment_model.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
IMDB Sentiment Analysis Model
A custom PyTorch LSTM model for predicting sentiment (positive/negative) from IMDB movie reviews.
Sentiment Analysis Model
Model Overview: This application is Sentiment Analysis Model built with PyTorch. The model is part of the Artificial Intelligence (AI)and Machine Learning(ML) fields, specifically focusing on Natural Language Processing (NLP). Data Source: The model is trained using the IMDB Dataset of 50K Movie Reviews, which contains movie reviews used for sentiment analysis. "https://www.kaggle.com/datasets/lakshmi25npathi/imdb-dataset-of-50k-movie-reviews"
Model Architecture:
The model is a Long Short-Term Memory (LSTM) neural network.
It uses an embedding layer to transform words into vector representations.
The LSTM layer processes the sequential input (the word embeddings).
The output from the last time step is passed through a dropout and a fully connected layer.
Model info
Created: 2025-03-05
Updated: 2025-03-05
View on Hugging FaceFiles (5)
Related models
More from XXuSea
No other models by this author.
More models
No task type declared.