From 76aa578ea0215a6a593b024c828c437dcb2cb647 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 24 Apr 2024 12:06:48 -0500
Subject: [PATCH] snapshot...
---
modules/machine_learning/classification.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/machine_learning/classification.py b/modules/machine_learning/classification.py
index 68bfed63..78c553a0 100644
--- a/modules/machine_learning/classification.py
+++ b/modules/machine_learning/classification.py
@@ -16,7 +16,7 @@ def get_csv_as_dataframe(csv_file):
return icing_df
-def get_train_test_data(data_frame, normalize=True):
+def get_train_test_data(data_frame, standardize=True):
icing_df = data_frame
# The independent variables we want to use:
params = ['cld_geo_thick', 'cld_temp_acha', 'conv_cloud_fraction', 'supercooled_cloud_fraction', 'cld_reff_dcomp',
@@ -32,7 +32,7 @@ def get_train_test_data(data_frame, normalize=True):
print(icing_df.shape)
x = np.asarray(icing_df[params])
- if normalize:
+ if standardize:
x = preprocessing.StandardScaler().fit(x).transform(x)
y = np.asarray(icing_df['icing_intensity'])
y = np.where(y == -1, 0, y)
--
GitLab