How to use amarshiv86/sentiment-analysis-imdb-model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="amarshiv86/sentiment-analysis-imdb-model")
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("amarshiv86/sentiment-analysis-imdb-model", dtype="auto")
5b986d3 9ad733c 5b986d3
1
2
3
4
5
6
7
{ "accuracy": 0.895, "f1": 0.8943, "roc_auc": 0.9598, "precision": 0.881, "recall": 0.908 }
