更新 function.py

function.py

@@ -48,14 +48,19 @@ def load_data(filepath):
     # 清洗数据，去除无法转换的值
     data = np.where(data == '?', '0', data)  # 将问号替换为0或其他合适的值
 
-    X = data[:, :-1].astype(float)  # 特征数据
+    # 确保所有数据都可以转换为浮点数
+    try:
+        X = data[:, :-1].astype(float)  # 特征数据
+    except ValueError as e:
+        print(f"特征数据转换错误: {e}")
+        return None, None
+
     y = data[:, -1]  # 标签数据
 
     # 将标签转换为 -1 和 1
     y = np.where(y == '1', 1, -1)  # 假设标签为 '1' 表示死亡，其他为存活
     return X, y
 
-
 if __name__ == "__main__":
     # 以下代码可用于测试，可自行随意修改
     # 加载训练集