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"
}