snapshot...

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()