ml/models/stutter/test_report.json

{
  "data_dir": "data/metadata/hybrid_dataset",
  "out_dir": "ml/models/stutter",
  "epochs": 3,
  "seed": 42,
  "base_model": "facebook/wav2vec2-base",
  "binary": true,
  "focal_gamma": 2.0,
  "n_train": 7486,
  "n_val": 1174,
  "n_test": 1666,
  "final_holdout": {
    "test_loss": 0.026271456852555275,
    "test_accuracy": 0.7160864345738295,
    "test_precision": 0.7182236788723905,
    "test_recall": 0.7172049523727114,
    "test_macro_f1": 0.7159307651129131,
    "test_runtime": 69.8198,
    "test_samples_per_second": 23.861,
    "test_steps_per_second": 2.993
  }
}