fix: fix some bugs (ensemble output, HPO, model tuning) (#648)

This commit is contained in:
Yuante Li
2025-02-27 18:28:13 +08:00
committed by GitHub
parent 2bceddf8a3
commit 9c8182acfe
3 changed files with 13 additions and 9 deletions
@@ -68,6 +68,8 @@ for key in val_preds_dict.keys():
else:
print(f"Model {key} test predictions (test_preds_dict[key]) shape: {test_preds_dict[key].shape}")
print(f"val_y.shape: {val_y.shape}" if not isinstance(val_y, list) else f"val_y(list)'s length: {len(val_y)}")
# Run ensemble
final_pred = ensemble_workflow(test_preds_dict, val_preds_dict, val_y)