更新 function.py

function.py

@@ -124,8 +124,17 @@ def load_data(filepath):
             if np.isnan(row[i]):
                 row[i] = mean_val
     
+    # 合并特征23的值2和3为0
+    for row in data:
+        if row[22] == 2 or row == 3:  # 假设特征23是第23个元素（索引为22）
+            row[22] = 0
+    
+    # 移除不需要的特征
+    # 假设特征从0开始编号，那么我们需要移除特征3（索引为2）和特征24~28（索引为23~27）
+    data = [row[:2] + row[3:23] + row[28:] for row in data]
+    
     # 分离特征和标签
-    X = [row[:-1] for row in data]
+    X = [row[:-1] for row in data]  # 特征现在是从0到21，然后是23到27
     y = [row[-1] for row in data]
     
     return np.array(X), np.array(y)