更新 function.py
Gitea Actions Experiment Test / Experiment-Actions (push) Successful in 32s

This commit is contained in:
2024-11-13 15:49:04 +08:00
parent 889cf74668
commit ff29b83efc
+2 -8
View File
@@ -145,11 +145,8 @@ def k_fold_cross_validation(X, y, k=5, learning_rate=0.001, iterations=10000, ba
if __name__ == "__main__": if __name__ == "__main__":
# 加载训练集 # 加载训练集
try:
X_train, y_train = load_data("./data/horse-colic.data") X_train, y_train = load_data("./data/horse-colic.data")
except FileNotFoundError as e:
print(e)
exit()
# 数据归一化处理 # 数据归一化处理
scaler = StandardScaler() scaler = StandardScaler()
@@ -160,11 +157,8 @@ if __name__ == "__main__":
print(f"K折交叉验证的平均准确率: {avg_accuracy:.4f}") print(f"K折交叉验证的平均准确率: {avg_accuracy:.4f}")
# 加载测试集 # 加载测试集
try:
X_test, y_test = load_data("./data/horse-colic.test") X_test, y_test = load_data("./data/horse-colic.test")
except FileNotFoundError as e:
print(e)
exit()
# 测试集也需要进行相同的归一化 # 测试集也需要进行相同的归一化
X_test = scaler.transform(X_test) # 使用相同的Scaler进行测试集的标准化 X_test = scaler.transform(X_test) # 使用相同的Scaler进行测试集的标准化