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

fix: fix a bug in ensemble test script (#713)

Browse Source 
* fix a bug in ensemble test script

* fix

* add a comment
This commit is contained in:
Yuante Li
2025-03-20 17:35:39 +08:00
committed by GitHub
parent 613121ca45
commit 0624ed180f
1 changed files with 2 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
+2 -1
View File
@@ -22,10 +22,11 @@ def print_preds_info(model_name, data_type, preds):
else:
print(f"Model {model_name} {data_type} predictions shape: {preds.shape}")
print("Showing a preview of the predictions (first few entries only):")
if isinstance(preds, (pd.DataFrame, pd.Series)):
print(preds.head())
elif isinstance(preds, (np.ndarray, torch.Tensor, tf.Tensor)):
print(pd.DataFrame(preds).head())
print(preds[:2])
elif isinstance(preds, list):
print(pd.DataFrame(preds[:5]))
else: