From 1437fce359d8b9c2dcc964f161d1183370d2fac3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=B4=E6=9D=A8=E6=B4=8B?= <2445091977@qq.com>
Date: Sat, 2 Nov 2024 21:58:28 +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 | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/function.py b/function.py
index 471bad3..3b1d494 100644
--- a/function.py
+++ b/function.py
@@ -37,7 +37,7 @@ def load_data(filepath):
             continue
         y.append(1 if data[22] in ['2', '3'] else 0)  # 假设标签是 '2' 或 '3' 为正类
         data_25 = data[24]
-        data = [float(x) if x != '?' else 0 for x in data[:24] + data_25.split(',')]  # 处理缺失值
+        data = [0 if x == '?' else float(x) for x in data[:24] + data_25.split(',')]  # 处理缺失值
         data_list.append(data)
     X = np.array(data_list)
     return X, np.array(y)
@@ -60,5 +60,5 @@ def main():
     accuracy_test = np.mean(y_pred_test == y_test)
     print(f"Test Accuracy: {accuracy_test:.2f}")
 
-if __name__.split("__main__"):
+if __name__ == "__main__":
     main()
\ No newline at end of file