From 27a58947789d1f52157d22e9462876016cefd996 Mon Sep 17 00:00:00 2001
From: Ulysses-ui <3233559397@qq.com>
Date: Sun, 3 Nov 2024 15:40:27 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/function.py b/function.py
index e24c5d2..25f3901 100644
--- a/function.py
+++ b/function.py
@@ -1,5 +1,5 @@
 import numpy as np
-from sklearn.preprocessing import StandardScaler
+from sklearn.preprocessing import StandardScaler, PolynomialFeatures
 
 id = "202210423124"
 name = "刘子豪"
@@ -81,9 +81,14 @@ if __name__ == "__main__":
     scaler = StandardScaler()
     X_train = scaler.fit_transform(X_train)
 
+    # 使用 PolynomialFeatures 生成多项式特征
+    poly = PolynomialFeatures(degree=2, include_bias=False)
+    X_train_poly = poly.fit_transform(X_train)
+
     # 加载测试集
     X_test, y_test = load_data("./data/horse-colic.test")
     X_test = scaler.transform(X_test)  # 使用同一标准化参数
+    X_test_poly = poly.transform(X_test)  # 应用相同的多项式特征转换
 
     # 训练对率回归模型
     weights = train_logreg(X_train, y_train, learning_rate=0.01, iterations=1000, reg_lambda=0.01)