cy/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 38s
Details

Browse Source
This commit is contained in:
cy
2024-10-26 19:51:55 +08:00
parent f71c28d4f7
commit 6f445ce205
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
function.py
+7 -2
View File
@@ -135,9 +135,14 @@ def grid_search(X_train, y_train, X_test, y_test, param_grid):
for params in product(*param_grid.values()): for params in product(*param_grid.values()):
learning_rate, iterations, batch_size, regularization, lambda_reg = params learning_rate, iterations, batch_size, regularization, lambda_reg = params
# 训练模型
weights = train_logreg(X_train, y_train, learning_rate, iterations, batch_size, regularization, lambda_reg) weights = train_logreg(X_train, y_train, learning_rate, iterations, batch_size, regularization, lambda_reg)
# 计算测试集准确率
acc = accuracy(X_test, y_test, weights) acc = accuracy(X_test, y_test, weights)
print(f"尝试参数组合: {params}, 测试集准确率: {acc:.4f}")
if acc > best_accuracy: if acc > best_accuracy:
best_accuracy = acc best_accuracy = acc
best_params = params best_params = params
@@ -145,8 +150,8 @@ def grid_search(X_train, y_train, X_test, y_test, param_grid):
return best_params, best_accuracy return best_params, best_accuracy
if __name__ == "__main__": if __name__ == "__main__":
X_train, y_train = load_data("./horse-colic.data") X_train, y_train = load_data("./data/horse-colic.data")
X_test, y_test = load_data("./horse-colic.test") X_test, y_test = load_data("./data/horse-colic.test")
# 定义参数网格 # 定义参数网格
param_grid = { param_grid = {