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.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support