From fd27a88e6a1814f799c1bb83276008e997a759e7 Mon Sep 17 00:00:00 2001 From: cy <939601402@qq.com> Date: Sat, 2 Nov 2024 15:13:35 +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 | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/function.py b/function.py index 0acbd86..16424fd 100644 --- a/function.py +++ b/function.py @@ -87,18 +87,17 @@ def accuracy(X, y, weights): if __name__ == "__main__": X_train, y_train = load_data("./data/horse-colic.data") - X_train, X_val, y_train, y_val = train_test_split(X_train, y_train, test_size=0.2, random_state=42) + X_test, y_test = load_data("./data/horse-colic.test") kf = KFold(n_splits=5, shuffle=True, random_state=42) - accuracies = [accuracy(X_val_kf, y_val_kf, train_logreg(X_train_kf, y_train_kf, batch_size=32)) + accuracies = [accuracy(X_test_kf, y_test_kf, train_logreg(X_train_kf, y_train_kf, batch_size=32)) for train_index, val_index in kf.split(X_train) - for X_train_kf, X_val_kf in [(X_train[train_index], X_train[val_index])] - for y_train_kf, y_val_kf in [(y_train[train_index], y_train[val_index])]] + for X_train_kf, X_test_kf in [(X_train[train_index], X_train[test_index])] + for y_train_kf, y_test_kf in [(y_train[train_index], y_train[test_index])]] print(f"K折交叉验证得分: {accuracies}") print(f"平均准确率: {np.mean(accuracies):.4f}") - X_test, y_test = load_data("./data/horse-colic.test") final_weights = train_logreg(X_train, y_train, batch_size=32) test_accuracy = accuracy(X_test, y_test, final_weights) print(f"测试集的准确率: {test_accuracy:.4f}")