diff --git a/modules/machine_learning/classification.py b/modules/machine_learning/classification.py index 0d4c399330e844497b90b9f11f912865ba633397..f6f321b716acdb5cc15523da83ab39d2a9e14bc0 100644 --- a/modules/machine_learning/classification.py +++ b/modules/machine_learning/classification.py @@ -15,6 +15,12 @@ import sklearn.tree as tree from sklearn.tree import export_graphviz +# def analyze(dataFrame): +# no_icing_df = dataFrame[dataFrame['icing_intensity'] == -1] +# icing_df = dataFrame[dataFrame['icing_intensity'] >= 1] +# return no_icing_df, icing_df + + def plot_confusion_matrix(cm, classes, normalize=False, title='Confusion matrix', @@ -80,6 +86,8 @@ def get_feature_target_data(data_frame, standardize=True): 'cld_opd_dcomp', 'iwc_dcomp'] # Remove this column icing_df = icing_df.drop('lwc_dcomp', axis=1) + # Remove this column + icing_df = icing_df.drop('cld_geo_thick', axis=1) # Remove rows with NaN values # icing_df = icing_df.dropna()