From 00e4f8deae05f996400628a802e0009f025cbd19 Mon Sep 17 00:00:00 2001 From: junos Date: Thu, 13 Apr 2023 13:04:53 +0200 Subject: [PATCH] More numeric_only arguments. See 1d903f36294bd7077a3ddd088d199292cc072f4b for explanation. --- src/features/all_cleaning_individual/straw/main.py | 2 +- src/features/all_cleaning_overall/straw/main.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/features/all_cleaning_individual/straw/main.py b/src/features/all_cleaning_individual/straw/main.py index dee4aba9..6a10486e 100644 --- a/src/features/all_cleaning_individual/straw/main.py +++ b/src/features/all_cleaning_individual/straw/main.py @@ -120,7 +120,7 @@ def straw_cleaning(sensor_data_files, provider): esm_cols = features.loc[:, features.columns.str.startswith('phone_esm_straw')] if provider["COLS_VAR_THRESHOLD"]: - features.drop(features.std()[features.std() == 0].index.values, axis=1, inplace=True) + features.drop(features.std(numeric_only=True)[features.std(numeric_only=True) == 0].index.values, axis=1, inplace=True) fe5 = features.copy() diff --git a/src/features/all_cleaning_overall/straw/main.py b/src/features/all_cleaning_overall/straw/main.py index a94023dd..be9b4c53 100644 --- a/src/features/all_cleaning_overall/straw/main.py +++ b/src/features/all_cleaning_overall/straw/main.py @@ -146,7 +146,7 @@ def straw_cleaning(sensor_data_files, provider, target): # (5) REMOVE COLS WHERE VARIANCE IS 0 if provider["COLS_VAR_THRESHOLD"]: - features.drop(features.std()[features.std() == 0].index.values, axis=1, inplace=True) + features.drop(features.std(numeric_only=True)[features.std(numeric_only=True) == 0].index.values, axis=1, inplace=True) graph_bf_af(features, "6variance_drop")