File size: 779 Bytes
c30a0c0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
26
27
28
29
30
31
{
  "model_type": "bart_with_classifier",
  "base_model": "facebook/bart-large",
  "num_labels": 1884,
  "dropout": 0.1,
  "task": "multi_label_classification",
  "domain": "bibliographic_classification",
  "language": "de",
  "classification_system": "BK (Basisklassifikation)",
  "performance": {
    "subset_accuracy": 0.257,
    "mcc": 0.498,
    "f1_micro": 0.479,
    "f1_macro": 0.214,
    "precision_micro": 0.661,
    "recall_micro": 0.376,
    "precision_macro": 0.338,
    "recall_macro": 0.175
  },
  "training_details": {
    "approach": "two_stage_fine_tuning",
    "stage1_epochs": 15,
    "stage2_epochs": 15,
    "batch_size": 64,
    "learning_rate": 2e-05,
    "dataset_size": 250831,
    "train_split": 0.7,
    "val_split": 0.15,
    "test_split": 0.15
  }
}