更新 function.py

2024-10-31 15:26:27 +08:00
parent f17a18e183
commit 9e13a90ce5
1 changed files with 3 additions and 0 deletions
@@ -117,6 +117,9 @@ def load_data(filepath):
    data = [line.strip().split() for line in lines]
    data = [list(map(lambda x: float(x) if x != '?' else np.nan, row)) for row in data]
    
+    # 移除类别标签缺失的样本
+    data = [row for row in data if row[-1] != np.nan]
+
    # 使用“0”填充缺失值
    for row in data:
        for i in range(len(row)):