From 1792bc1796f2fe121d511b075dd84f3c5c4aee4d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 29 Apr 2024 15:15:58 -0500
Subject: [PATCH] snapshot...
---
modules/machine_learning/classification.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/modules/machine_learning/classification.py b/modules/machine_learning/classification.py
index f6f321b7..87de1486 100644
--- a/modules/machine_learning/classification.py
+++ b/modules/machine_learning/classification.py
@@ -15,10 +15,10 @@ 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 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,
@@ -82,7 +82,7 @@ def get_feature_target_data(data_frame, standardize=True):
icing_df = data_frame
# The independent variables (features) we want to use:
- params = ['cld_geo_thick', 'cld_temp_acha', 'conv_cloud_fraction', 'supercooled_cloud_fraction', 'cld_reff_dcomp',
+ params = ['cld_temp_acha', 'conv_cloud_fraction', 'supercooled_cloud_fraction', 'cld_reff_dcomp',
'cld_opd_dcomp', 'iwc_dcomp']
# Remove this column
icing_df = icing_df.drop('lwc_dcomp', axis=1)
--
GitLab