Edit model card

Used the website categories from URL Classification Dataset [DMOZ] https://www.kaggle.com/datasets/shawon10/url-classification-dataset-dmoz and a zero-shot-classifier on c4 sample to make a training dataset. Only a subset of the linked dataset was used.

Model Trained Using AutoTrain

  • Problem type: Text Classification

Validation Metrics

loss: 0.45181071758270264

f1_macro: 0.9103583954158208

f1_micro: 0.9108333333333334

f1_weighted: 0.9103583954158205

precision_macro: 0.9123282026272069

precision_micro: 0.9108333333333334

precision_weighted: 0.9123282026272069

recall_macro: 0.9108333333333334

recall_micro: 0.9108333333333334

recall_weighted: 0.9108333333333334

accuracy: 0.9108333333333334

Downloads last month
9
Safetensors
Model size
435M params
Tensor type
F32
·
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.

Dataset used to train KnutJaegersberg/website-classifier