diff --git a/modules/machine_learning/classification.py b/modules/machine_learning/classification.py index 506db5e3b00fd30d31f3b2742dab46573e823569..72f2227eb1cfdf2eff246c69b316d1f73ee69f67 100644 --- a/modules/machine_learning/classification.py +++ b/modules/machine_learning/classification.py @@ -83,7 +83,7 @@ def get_feature_target_data(data_frame, standardize=True): # The independent variables (features) we want to use: params = ['cld_temp_acha', 'conv_cloud_fraction', 'supercooled_cloud_fraction', 'cld_reff_dcomp', - 'cld_opd_dcomp', 'iwc_dcomp'] + 'cld_opd_dcomp', 'iwc_dcomp', 'cld_cwp_dcomp'] # Remove this column icing_df = icing_df.drop('lwc_dcomp', axis=1) # Remove this column