File size: 602 Bytes
df94406
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "model_type": "wav2vec2_classifier",
  "base_model": "facebook/wav2vec2-base",
  "num_classes": 2,
  "class_names": ["background", "switch"],
  "sample_rate": 16000,
  "max_length": 80000,
  "feature_size": 768,
  "classifier_config": {
    "hidden_size": 256,
    "dropout": 0.3,
    "num_layers": 2
  },
  "training_config": {
    "epochs": 15,
    "learning_rate": 1e-4,
    "batch_size": 4,
    "optimizer": "AdamW",
    "loss_function": "CrossEntropyLoss"
  },
  "performance": {
    "accuracy": 1.0,
    "precision": 1.0,
    "recall": 1.0,
    "f1_score": 1.0
  }
}