gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-28 07:57:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

add a check for the name of ensemble's score in scores.csv (#614)

Browse Source
This commit is contained in:
Yuante Li
2025-02-18 22:24:11 +08:00
committed by GitHub
parent 8bdccb59d7
commit f200320b3f
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/ensemble/eval_tests/ensemble_test.txt
+6 -1
View File
@@ -93,8 +93,13 @@ model_set_in_scores = set(score_df.index)
for model in {{model_names}}:
if model not in model_set_in_scores:
print(f"\nModel {model} is not evaluated in the scores.csv.")
print("Please check the format of the scores dataframe and ignore the magnitude of its values:")
print("Please check the format of the scores dataframe:")
print(score_df)
if "ensemble" not in model_set_in_scores:
print("\nError: Ensemble results are missing or incorrectly saved in `scores.csv`. "
"Ensure that the ensemble's performance is recorded as 'ensemble' in the file.")
print("Ensemble test end.")
print(f"Final prediction shape: {final_pred.shape}")