更新 function.py

function.py

@@ -6,19 +6,31 @@ name = "尹家立"
 
 def load_data(filepath):
     """
-    从文件中加载数据，若不自行正确实现，则无法通过评测，**不记录成绩**。
+    从文件中加载数据，并处理缺失值。
+
     Args:
     - filepath (str): 数据文件路径。
+
     Returns:
     - np.array: 特征数据。
     - np.array: 标签数据。
     """
-    data = np.loadtxt(filepath, delimiter=' ', dtype=float)
-    X = data[:, :-1]  # 特征
-    y = data[:, -1]   # 标签
+    # 读取数据，替换缺失值 '?' 为 NaN
+    data = np.genfromtxt(filepath, delimiter=' ', dtype=float, missing_values='?', filling_values=np.nan)
+
+    # 将 NaN 替换为列均值
+    nan_indices = np.isnan(data)
+    col_means = np.nanmean(data, axis=0)
+    data[nan_indices] = np.take(col_means, np.where(nan_indices)[1])
+
+    # 将最后一列作为标签
+    X = data[:, :-1]
+    y = data[:, -1]
+    
     return X, y
 
 
+
 def train_svm(X, y, learning_rate=0.01, iterations=1000):
     """
     支持向量机训练函数。