From cbda375486d60780fde72ad88adb3391ffd38660 Mon Sep 17 00:00:00 2001
From: wujunjie <297081150@qq.com>
Date: Sun, 6 Oct 2024 18:33:48 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/function.py b/function.py
index 687ee43..fa285a4 100644
--- a/function.py
+++ b/function.py
@@ -80,8 +80,14 @@ def load_data(file_path):
             values = line.strip().split("\t")
             features.append([float(x) for x in values[:-1]])
             labels.append(int(values[-1]))
-
-    return np.array(features), np.array(labels)
+    features = np.array(features)  
+    labels = np.array(labels)  
+  
+    # 归一化特征值  
+    scaler = StandardScaler()  
+    features_scaled = scaler.fit_transform(features)  
+  
+    return features_scaled, labels, scaler  # 返回归一化后的特征和缩放器对象
 
 
 if __name__ == "__main__":