From d54e5339b3b61794369fa2c2f7a47a9f54efab1d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 26 Apr 2024 15:43:59 -0500
Subject: [PATCH] snapshot...
---
modules/machine_learning/classification.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/modules/machine_learning/classification.py b/modules/machine_learning/classification.py
index 0d4c3993..f6f321b7 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()
--
GitLab