Ulysses-ui/Experiment-LogReg
1
1
Fork 0
generated from LufsX/Experiment-LogReg
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 function.py
Gitea Actions Experiment Test / Experiment-Actions (push) Successful in 28s
Details

Browse Source
This commit is contained in:
Ulysses-ui
2024-11-03 15:40:27 +08:00
parent ea6d4ebca5
commit 27a5894778
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
function.py
+6 -1
View File
@@ -1,5 +1,5 @@
import numpy as np import numpy as np
from sklearn.preprocessing import StandardScaler from sklearn.preprocessing import StandardScaler, PolynomialFeatures
id = "202210423124" id = "202210423124"
name = "刘子豪" name = "刘子豪"
@@ -81,9 +81,14 @@ if __name__ == "__main__":
scaler = StandardScaler() scaler = StandardScaler()
X_train = scaler.fit_transform(X_train) 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, y_test = load_data("./data/horse-colic.test")
X_test = scaler.transform(X_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) weights = train_logreg(X_train, y_train, learning_rate=0.01, iterations=1000, reg_lambda=0.01)