更新 function.py

2024-11-14 16:48:57 +08:00
parent 09f9f6063c
commit de1f52e5a6
1 changed files with 4 additions and 2 deletions
@@ -88,8 +88,10 @@ def load_data(filepath):
    data = []
    with open(filepath, 'r') as file:
        for line in file:
-            if "?" not in line:  # 去除包含缺失值的数据行
-                data.append([float(x) for x in line.strip().split()])
+            # 将 "?" 替换为 "0"，然后将整行数据转换为浮点数
+            line = line.strip().replace("?", "0")
+            data.append([float(x) for x in line.split()])
+    
    data = np.array(data)
    X = data[:, :-1]  # 所有列，除最后一列
    y = data[:, -1]   # 最后一列