更新 function.py

function.py

@@ -15,18 +15,16 @@ import numpy as np
 """
 id = "202210423208"
 name = "刘义天"
-
-
-# 这里实现数据加载函数
+# 加载数据并添加特征交互项
 def load_data(filepath):
     """
-    从文件中加载数据，若不自行正确实现，则无法通过评测，**不记录成绩**
+    从文件中加载数据并生成特征交互项。
 
     Args:
     - filepath (str): 数据文件路径
 
     Returns:
-    - X (numpy array): 特征数据
+    - X (numpy array): 包含特征交互项的特征数据
     - y (numpy array): 标签数据
     """
     # 读取数据，指定分隔符为空格，缺失值符号为 '?'
@@ -45,8 +43,20 @@ def load_data(filepath):
     X = data[:, :-1]
     y = data[:, -1]
     
+    # 生成特征交互项
+    n = X.shape[1]
+    interaction_features = []
+    for i in range(n):
+        for j in range(i + 1, n):
+            interaction_features.append(X[:, i] * X[:, j])
+    
+    # 将交互特征添加到原特征集
+    X = np.hstack([X] + interaction_features)
+    
     return X, y
-# 这里实现对率回归训练函数
+
+
+# 对率回归训练函数
 def train_logreg(X, y, learning_rate=0.01, num_iterations=1000, regularization_strength=0):
     """
     对率回归训练函数。
@@ -68,7 +78,9 @@ def train_logreg(X, y, learning_rate=0.01, num_iterations=1000, regularization_s
         gradient = (1 / m) * np.dot(X.T, (predictions - y)) + regularization_strength * weights
         weights -= learning_rate * gradient
     return weights
-# 这里实现对率回归预测函数
+
+
+# 对率回归预测函数
 def predict_logreg(inX, weights):
     """
     对率回归预测函数。
@@ -82,6 +94,7 @@ def predict_logreg(inX, weights):
     """
     probability = 1 / (1 + np.exp(-np.dot(inX, weights)))
     return 1 if probability >= 0.5 else 0
+
 if __name__ == "__main__":
     # 以下代码可用于测试，可自行随意修改
     # 加载训练集