train_results.json

{
    "epoch": 3.0,
    "train_accuracy": 0.6,
    "train_auc_class_0": 0.7,
    "train_auc_class_1": 0.96,
    "train_auc_class_2": 0.8200000000000001,
    "train_auc_overall": 0.8266666666666667,
    "train_loss": 0.9045717120170593,
    "train_runtime": 1.6579,
    "train_samples_per_second": 9.047,
    "train_steps_per_second": 0.603
}