metadata
license: apache-2.0
tags:
- keras
- tensorflow
- text-classification
- sentiment-analysis
library_name: tensorflow
task: text-classification
Conv1D Sentiment Analysis Model
This is a Keras Conv1D model trained on the IMDB movie review dataset for binary sentiment classification.
- Labels: 0 = Negative, 1 = Positive
- Preprocessing: TextVectorization layer included inside the model
- Architecture: Embedding -> Conv1D -> GlobalMaxPooling -> Dense layers
- Embeddings: Can use pretrained embeddings or trainable embeddings
Usage Example
from tensorflow.keras.models import load_model
model = load_model("sentiment_model.keras")
pred = model.predict(["This movie was amazing"])
print(pred)
This model was trained in Google Colab.