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 a57e625b17
commit 68bf5665f0
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)
@@ -39,8 +39,7 @@ model_coder:
--------- Feature Engineering Code ---------
{{ feature_code }}
2. You should avoid using logging module to output information in your generated code, and instead use the print() function.
3. You can decide whether to use AutoML based on the characteristics of the task.
4. If the model can both be implemented by PyTorch and Tensorflow, please use pytorch for broader compatibility.
3. If the model can both be implemented by PyTorch and Tensorflow, please use pytorch for broader compatibility.
## Output Format
{% if out_spec %}