From ae5349974227dd15075d3cca8762c168c75963d9 Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Fri, 22 Dec 2023 09:51:57 +0000 Subject: [PATCH] add keep only non negative carbon change to carbon model --- .../src/pipeline/configs/feature_processor_logic.py | 6 ++++++ modules/ml-pipeline/src/pipeline/example.py | 0 2 files changed, 6 insertions(+) create mode 100644 modules/ml-pipeline/src/pipeline/example.py diff --git a/modules/ml-pipeline/src/pipeline/configs/feature_processor_logic.py b/modules/ml-pipeline/src/pipeline/configs/feature_processor_logic.py index 78c29a9..bce32b6 100644 --- a/modules/ml-pipeline/src/pipeline/configs/feature_processor_logic.py +++ b/modules/ml-pipeline/src/pipeline/configs/feature_processor_logic.py @@ -23,6 +23,11 @@ def keep_negative_heat_change(df): return df +def keep_non_negative_carbon_ending(df): + df = df[df["CARBON_ENDING"] > 0] + return df + + def keep_negative_carbon_change(df): df = df[df["CARBON_CHANGE"] < 0] return df @@ -68,6 +73,7 @@ business_logic = { "keep_negative_carbon_change": keep_negative_carbon_change, "remove_top_1_percent_heat_demand": remove_top_1_percent_heat_demand, "remove_top_1_percent_carbon": remove_top_1_percent_carbon, + "keep_non_negative_carbon_ending": keep_non_negative_carbon_ending # "remove_starting_columns": remove_starting_columns # "keep_ENDING_COLUMNS": keep_ending_columns } diff --git a/modules/ml-pipeline/src/pipeline/example.py b/modules/ml-pipeline/src/pipeline/example.py new file mode 100644 index 0000000..e69de29