File size: 600 Bytes
219d96c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
{
"model_name": "IntentAnalyzer",
"base_model": "distilbert-base-uncased",
"task": "multi_label_text_classification",
"training_examples": 858,
"validation_examples": 184,
"test_examples": 184,
"total_examples": 1226,
"intent_categories": [
"trolling",
"dismissive",
"manipulative",
"emotionally_reactive",
"constructive",
"unclear"
],
"num_labels": 6,
"f1_macro": 0.77,
"training_epochs": 4,
"batch_size": 16,
"learning_rate": 2e-05,
"max_length": 128,
"architecture": "DistilBERT + Linear Classifier",
"loss_function": "BCEWithLogitsLoss"
} |