diff --git a/function.py b/function.py index 0f1346c..7efd0a1 100644 --- a/function.py +++ b/function.py @@ -137,8 +137,8 @@ if __name__ == "__main__": X_test, y_test = load_data("./horse-colic.test") # 训练对率回归模型 - weights = train_logreg(X_train, y_train, learning_rate=0.2, iterations=200, batch_size=32, regularization='L1', - lambda_reg=0.15) + weights = train_logreg(X_train, y_train, learning_rate=0.25, iterations=1500, batch_size=16, regularization='L2', + lambda_reg=0.5) # 计算并输出准确率 acc = accuracy(X_test, y_test, weights) print(f"模型的准确率: {acc:.2f}")