更新 function.py

function.py

@@ -60,13 +60,11 @@ def predict_logreg(inX, weights):
 def load_data(filepath):
     data = []
     labels = []
-    line_number = 0  # 初始化行号计数器
 
     with open(filepath, 'r') as f:
-        for line in f:
-            line_number += 1  # 每读取一行，行号加1
-            # 跳过第三行和第24到28行
-            if line_number == 3 or line_number in range(24, 29):
+        for line_number, line in enumerate(f):
+            # 忽略第三行和第24到28行
+            if line_number == 2 or (23 <= line_number <= 27):
                 continue
             
             values = line.strip().replace('?', '0').split()
@@ -96,13 +94,11 @@ def load_data(filepath):
 
     return df.to_numpy(), labels
 
-
 def preprocess_data(X):
     # 特征标准化
     scaler = StandardScaler()
     return scaler.fit_transform(X)
 
-
 if __name__ == "__main__":
     # 加载数据集
     X, y = load_data("./data/horse-colic.data")