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
- 74