test_metrics.json

{'test_loss': 0.662712037563324, 'test_r2': 0.755, 'test_spearmanr': 0.8656511076800086, 'test_runtime': 22.3756, 'test_samples_per_second': 319.723, 'test_steps_per_second': 20.022}