更新 function.py

function.py

@@ -90,12 +90,23 @@ def load_data(filepath):
             values = line.strip().split()
             # 检查是否存在 '?' 并替换为 0 或者跳过这些行
             if '?' in values:
-                # 可以选择跳过整行或者替换为0，这里选择跳过整行
                 continue
             else:
                 # 将标签和特征转换为浮点数
                 y.append(int(values[-1]))
-                X.append([float(x) if x != '?' else 0.0 for x in values[:-1]])
+                X.append([float(x) for x in values[:-1]])
+    
+    # 将特征数据转换为 numpy 数组
+    X_array = np.array(X)
+    
+    # 计算每列的平均值，用于替换 '?'
+    column_means = np.mean(X_array, axis=0)
+    
+    # 替换 '?' 为每列的平均值
+    for i in range(len(X)):
+        for j in range(len(X[i])):
+            if X[i][j] == '?':
+                X[i][j] = column_means[j]
     
     return np.array(X), np.array(y)