ameerazam08
/

autotrain-imdb-1166543179

Text Classification

Trained with AutoTrain

Inference Endpoints

Model card Files Files and versions Community

Edit model card

YAML Metadata Error: "tags" must be an array

Model Trained Using AutoTrain

Problem type: Binary Classification
Model ID: 1166543179
CO2 Emissions (in grams): 0.4999789160111311

Validation Metrics

Loss: 0.19526566565036774
Accuracy: 0.9418
Precision: 0.9441093687173301
Recall: 0.9392
AUC: 0.9824502399999999
F1: 0.9416482855424103

Usage

You can use cURL to access this model:

$ curl -X POST -H "Authorization: Bearer YOUR_API_KEY" -H "Content-Type: application/json" -d '{"inputs": "I love AutoTrain"}' https://api-inference.huggingface.co/models/ameerazam08/autotrain-imdb-1166543179

Or Python API:

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained("ameerazam08/autotrain-imdb-1166543179", use_auth_token=True)

tokenizer = AutoTokenizer.from_pretrained("ameerazam08/autotrain-imdb-1166543179", use_auth_token=True)

inputs = tokenizer("I love AutoTrain", return_tensors="pt")

outputs = model(**inputs)

Downloads last month: 20

Inference Examples

Text Classification

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train ameerazam08/autotrain-imdb-1166543179