更新 function.py

2024-11-02 16:07:42 +08:00
parent d56dda8fbb
commit c1e69211b6
1 changed files with 3 additions and 2 deletions
@@ -69,9 +69,10 @@ def load_data(filepath):
    data, labels = [], []
    with open(filepath, 'r') as f:
        for i, line in enumerate(f):
-            if i in [2] or (24 <= i <= 28):  # 剔除第3行和第24-28行
-                continue
            line = line.strip().split()
+            # 只处理数据行，跳过第3行和第24-28行
+            if i in [2] or (24 <= i <= 28):
+                continue
            if len(line) >= 1 and line[-1].isdigit():
                line_data = [0 if x == '?' else float(x) for x in line[:-1]]
                data.append(line_data)