From 3a3adf5b8674d94c7ebac81e82eee2ba15803a32 Mon Sep 17 00:00:00 2001 From: zhangchenghong <1598349064@qq.com> Date: Thu, 31 Oct 2024 17:06:50 +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 | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/function.py b/function.py index 11e5d50..68bab97 100644 --- a/function.py +++ b/function.py @@ -115,8 +115,12 @@ if __name__ == "__main__": # 加载测试集 X_test, y_test = load_data("./data/horse-colic.test") - # 训练对率回归模型 + # 进行 K 折交叉验证 k = 5 # 设置折数 - accuracy = cross_validate(X_train, y_train, k=k, learning_rate=0.01, iterations=5000) + cv_accuracy = cross_validate(X_train, y_train, k=k, learning_rate=0.01, iterations=1000) + print(f"{k}-折交叉验证的平均准确率: {cv_accuracy:.2f}") - print(f"{k}-折交叉验证的平均准确率: {accuracy:.2f}") + # 测试单个数据 + weights = train_logreg(X_train, y_train, learning_rate=0.01, iterations=1000) + y_pred = predict_logreg(X_test[11], weights) + print(f"预测结果: {y_pred}, 实际结果: {y_test[11]}")