Model Trained Using AutoTrain

  • Problem type: Binary Classification
  • Model ID: 3390592983
  • CO2 Emissions (in grams): 0.0090

Validation Metrics

  • Loss: 0.014
  • Accuracy: 0.997
  • Precision: 0.998
  • Recall: 0.997
  • AUC: 1.000
  • F1: 0.998

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/davanstrien/autotrain-dataset-mentions-3390592983

Or Python API:

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained("davanstrien/autotrain-dataset-mentions-3390592983", use_auth_token=True)

tokenizer = AutoTokenizer.from_pretrained("davanstrien/autotrain-dataset-mentions-3390592983", use_auth_token=True)

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

outputs = model(**inputs)
Downloads last month
19
Inference Examples
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.

Model tree for davanstrien/autotrain-dataset-mentions-3390592983

Finetuned
(99)
this model

Dataset used to train davanstrien/autotrain-dataset-mentions-3390592983