BinaryMans
/

my-first-ai-model

Text Classification

binary-classification

Model card Files Files and versions

You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

🚀 My First AI Model

✅ This is my first text classification model, fine-tuned using distilbert-base-uncased on a binary sentiment dataset.

Task: Sentiment Analysis (Positive/Negative)
Labels:
- LABEL_1: Positive
- LABEL_0: Negative

📊 Training Data

Custom CSV dataset with two columns:

text: The input text.
label: 0 for Negative, 1 for Positive.

🧪 Metrics

Metric	Value
Training Loss	~0.71
Epochs	2

🚀 How to use

With 🤗 `transformers` (locally):

from transformers import pipeline

clf = pipeline("text-classification", model="BinaryMans/my-first-ai-model")
clf("I love this product!")

Downloads last month: -

Safetensors

Model size

67M params

Tensor type

F32

·