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)
@@ -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 %}