all_results.json

{
    "epoch": 2.7203482045701852,
    "eval_loss": 0.006370748393237591,
    "eval_mae_per_cat": [
    0.025086652487516403,
    0.03752453625202179,
    0.02304081991314888,
    0.013712005689740181,
    0.030296988785266876,
    0.006659933365881443
    ],
    "eval_mean_mae": 0.022720156237483025,
    "eval_mean_pearson": 0.9476440191268921,
    "eval_mean_r2": 0.7471505999565125,
    "eval_mean_rmse": 0.07224167883396149,
    "eval_pearson_per_cat": [
    0.9419375658035278,
    0.9502590894699097,
    0.9368577599525452,
    0.948063850402832,
    0.9611018300056458,
    NaN
    ],
    "eval_r2_per_cat": [
    0.8866956233978271,
    0.8990016579627991,
    0.8755217790603638,
    0.8987970352172852,
    0.9228875637054443,
    0.0
    ],
    "eval_rmse_per_cat": [
    0.08463889360427856,
    0.11158231645822525,
    0.08321022242307663,
    0.04930051788687706,
    0.09579125046730042,
    0.008926868438720703
    ],
    "eval_runtime": 104.2098,
    "eval_samples_per_second": 30.247,
    "eval_steps_per_second": 3.781
}