dong/Experiment-LogReg
1
0
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 31s
Details

Browse Source
This commit is contained in:
dong
2024-11-01 00:17:46 +08:00
parent 6936ea4ee5
commit a886e9d94f
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
function.py
+6 -6
View File
@@ -1,6 +1,7 @@
import numpy as np import numpy as np
import pandas as pd import pandas as pd
from sklearn.preprocessing import StandardScaler from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import train_test_split
id = "202210423130" id = "202210423130"
@@ -89,13 +90,12 @@ def preprocess_data(X):
if __name__ == "__main__": if __name__ == "__main__":
# 加载训练集 # 加载数据集
X_train, y_train = load_data("./data/horse-colic.data") X, y = load_data("./data/horse-colic.data")
X_train = preprocess_data(X_train) X = preprocess_data(X)
# 加载测试集 # 使用 train_test_split 打乱并随机抽取数据
X_test, y_test = load_data("./data/horse-colic.test") X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
X_test = preprocess_data(X_test)
# 训练对率回归模型 # 训练对率回归模型
weights = train_logreg(X_train, y_train, learning_rate=0.1, iterations=1000, lambda_reg=0.01) weights = train_logreg(X_train, y_train, learning_rate=0.1, iterations=1000, lambda_reg=0.01)