更新 function.py

2024-10-31 15:34:22 +08:00
parent e55b05db9b
commit 43a7104511
1 changed files with 4 additions and 1 deletions
@@ -113,6 +113,8 @@ if __name__ == "__main__":

    # 数据预处理
    X, y = preprocess_data(X, y)
+    X_mean = np.mean(X, axis=0)  # 保存训练数据的均值
+    X_std = np.std(X, axis=0)    # 保存训练数据的标准差

    # 尝试不同的正则化强度
    l2_penalties = [0.01, 0.1, 1.0]
@@ -123,7 +125,8 @@ if __name__ == "__main__":

        # 加载测试数据并进行预处理
        X_test, y_test = load_data("./data/horse-colic.test")
-        X_test, y_test = preprocess_data(X_test, y_test)
+        X_test = (X_test - X_mean) / X_std  # 使用训练数据的均值和标准差进行标准化
+        X_test, y_test = preprocess_data_base(X_test, y_test)  # 应用其他预处理步骤
        
        # 评估模型准确率
        accuracy = evaluate_model(X_test, y_test, weights)