diff --git a/function.py b/function.py index 5a22a31..e46cdc4 100644 --- a/function.py +++ b/function.py @@ -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)