gpu-runtime-predictor / metrics.json
RajBhope's picture
Upload metrics.json with huggingface_hub
39cc498 verified
{
"gbr": {
"val_mse": 0.005296896676123843,
"val_rmse": 0.07277978205603423,
"val_mae": 0.03653335769226314,
"val_r2": 0.9923377189577787,
"val_mape_pct": 16.450425059017586,
"val_spearman": 0.926380485553136
},
"rf": {
"val_mse": 0.005240080574356092,
"val_rmse": 0.07238840082745364,
"val_mae": 0.03626183644611175,
"val_r2": 0.9924199068814038,
"val_mape_pct": 16.344032442474585,
"val_spearman": 0.927697311160747
},
"nn": {
"val_mse": 0.004724574275314808,
"val_rmse": 0.06873553866316033,
"val_mae": 0.03571900725364685,
"val_r2": 0.9931656122207642,
"val_mape_pct": 16.984268188476562,
"val_spearman": 0.9187410777298199
},
"ensemble": {
"val_mse": 0.004809145669223057,
"val_rmse": 0.06934800407526562,
"val_mae": 0.03530809874932828,
"val_r2": 0.9930432802556507,
"val_mape_pct": 16.32863143030719,
"val_spearman": 0.9272286600491021
},
"best_model": "NN",
"ensemble_weights": {
"gbr": 0.5,
"rf": 0.3,
"nn": 0.2
}
}