From ff29b83efc1861bd2872ac76efce5b4e45d0ec39 Mon Sep 17 00:00:00 2001 From: heart20003 <1789588961@qq.com> Date: Wed, 13 Nov 2024 15:49:04 +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 | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/function.py b/function.py index 709b667..5ae9dc7 100644 --- a/function.py +++ b/function.py @@ -145,11 +145,8 @@ def k_fold_cross_validation(X, y, k=5, learning_rate=0.001, iterations=10000, ba if __name__ == "__main__": # 加载训练集 - try: - X_train, y_train = load_data("./data/horse-colic.data") - except FileNotFoundError as e: - print(e) - exit() + X_train, y_train = load_data("./data/horse-colic.data") + # 数据归一化处理 scaler = StandardScaler() @@ -160,11 +157,8 @@ if __name__ == "__main__": print(f"K折交叉验证的平均准确率: {avg_accuracy:.4f}") # 加载测试集 - try: - X_test, y_test = load_data("./data/horse-colic.test") - except FileNotFoundError as e: - print(e) - exit() + X_test, y_test = load_data("./data/horse-colic.test") + # 测试集也需要进行相同的归一化 X_test = scaler.transform(X_test) # 使用相同的Scaler进行测试集的标准化