Skip to content

Commit

Permalink
fix a bug in ensemble module
Browse files Browse the repository at this point in the history
  • Loading branch information
WinstonLiyt committed Dec 30, 2024
1 parent 72bfa90 commit d8b5a4c
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -227,7 +227,7 @@ spec:
- Data loading (`load_data.py`).
- Feature engineering (`feat*.py`).
- Model workflow for training and testing (`model*.py`).
- Ensemble and decision-making (`ens.py`).
- Ensemble and decision-making (`ensemble.py`).
- Treat each component as a modular and callable Python function.
2. Dataset Splitting
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ workflow_coder:
{{ model_code }}
---------ensemble code---------
file: ens.py
file: ensemble.py
{{ ensemble_code }}
{% if latest_code %}
Expand Down
2 changes: 1 addition & 1 deletion rdagent/components/coder/data_science/workflow/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def develop_one_competition(competition: str):
)

tpl_ex_path = Path(__file__).resolve() / Path("rdagent/scenarios/kaggle/tpl_ex").resolve() / competition
injected_file_names = ["spec/workflow.md", "load_data.py", "feat01.py", "model01.py", "ens.py", "main.py"]
injected_file_names = ["spec/workflow.md", "load_data.py", "feat01.py", "model01.py", "ensemble.py", "main.py"]

workflowexp = FBWorkspace()
for file_name in injected_file_names:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@


# Ensemble
from ens import ens_and_decision
from ensemble import ens_and_decision

pred_binary = ens_and_decision([test_pred], [val_pred], validation_labels)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ The project should be organized into the following components:
1. **Data Loading** (`load_data.py`): A module responsible for loading and preprocessing raw data.
2. **Feature Engineering**(`feat*.py`): A module for transforming raw data into features suitable for model training.
3. **Model Workflow**(`model*.py`): A module that manages the training, validation, and testing of machine learning models.
4. **Ensemble and Decision Making**(`ens.py`): A module for combining predictions from multiple models and making final decisions.
4. **Ensemble and Decision Making**(`ensemble.py`): A module for combining predictions from multiple models and making final decisions.
5. **Workflow**(`main.py`): A script to put the above component together to get the final submission(`submission.csv`)

## Submission
Expand Down

0 comments on commit d8b5a4c

Please sign in to comment.