diff --git a/src/models/select_targets.py b/src/models/select_targets.py index 196cdcd1..4dc4a252 100644 --- a/src/models/select_targets.py +++ b/src/models/select_targets.py @@ -6,5 +6,6 @@ cleaned_sensor_features = pd.read_csv(snakemake.input["cleaned_sensor_features"] target_variable_name = snakemake.params["target_variable"] model_input = retain_target_column(cleaned_sensor_features, target_variable_name) +model_input.dropna(axis ="index", how="any", subset=["target"], inplace=True) model_input.to_csv(snakemake.output[0], index=False)