Skip to content
Snippets Groups Projects
Commit 9a45dc8b authored by Andri Joos's avatar Andri Joos :blush:
Browse files

fix dropping columns when removing unimportant predictors

parent ab6f7296
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
app/preprocessing/transform_dataset.py
+ 1
− 1
View file @ 9a45dc8b
...@@ -194,7 +194,7 @@ def _remove_unimportant_predictors(train_files: List[Path], all_files: List[Path ...@@ -194,7 +194,7 @@ def _remove_unimportant_predictors(train_files: List[Path], all_files: List[Path
df = pd.read_parquet(file) df = pd.read_parquet(file)
columns_to_drop = [col for col in df.columns if col not in columns_to_keep] columns_to_drop = [col for col in df.columns if col not in columns_to_keep]
df.drop(columns=columns_to_drop) df = df.drop(columns=columns_to_drop)
df.to_parquet(out_dir / file.name) df.to_parquet(out_dir / file.name)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment