generated from LufsX/Experiment-LogReg
This commit is contained in:
+2
-8
@@ -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()
|
||||
|
||||
|
||||
# 数据归一化处理
|
||||
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 = scaler.transform(X_test) # 使用相同的Scaler进行测试集的标准化
|
||||
|
||||
Reference in New Issue
Block a user