{
    "test_loss": 0.9019122123718262,
    "test_hs_f1": 0.6540642722117203,
    "test_hs_precision": 0.49570200573065903,
    "test_hs_recall": 0.9611111111111111,
    "test_tr_f1": 0.6102635228848822,
    "test_tr_precision": 0.4819277108433735,
    "test_tr_recall": 0.831758034026465,
    "test_ag_f1": 0.482573726541555,
    "test_ag_precision": 0.35405192761605037,
    "test_ag_recall": 0.7575757575757576,
    "test_macro_hs_f1_score": 0.548190168321794,
    "test_emr_no_gating": 0.41133333333333333,
    "test_emr": 0.4116666666666667,
    "test_macro_f1": 0.5823004841804504,
    "test_macro_precision": 0.4438938796520233,
    "test_macro_recall": 0.8501482605934143,
    "test_runtime": 6.346,
    "test_samples_per_second": 472.739,
    "test_steps_per_second": 29.625
}