Skip to content

Commit 00c6d0f

Browse files
committed
Merge branch 'main' into regression-random-state
2 parents 1757e45 + 25240a1 commit 00c6d0f

2 files changed

Lines changed: 3 additions & 0 deletions

File tree

docs/changes/82.feature.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Add regression traing with a reduced list of features.

src/eventdisplay_ml/models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -872,6 +872,8 @@ def train_regression(df, model_configs):
872872
# all-non-target feature set.
873873
profile = model_configs.get("feature_profile", "extended")
874874
x_cols = features.regression_feature_columns(df.columns, profile=profile)
875+
excluded_targets = set(model_configs["targets"])
876+
x_cols = [col for col in x_cols if col not in excluded_targets]
875877
_logger.info(f"Features ({len(x_cols)}): {', '.join(list(x_cols))}")
876878
model_configs["features"] = list(x_cols)
877879
targets = model_configs["targets"]

0 commit comments

Comments
 (0)