逻辑回归调用中出现警告

Question

from sklearn.linear_model import LogisticRegression

log_reg = LogisticRegression()
log_reg.fit(X_train, y_train)
log_reg.score(X_test, y_test)

老师好，本节当中逻辑回归调用中，视频中并不会出现警告，而在用新版sklearn运行时则老是会出现如图所示警告，请问这个问题是出在哪里呢？

Answer 1

liuyubobobo 2022-08-14 16:14:19

新版本下，在默认的 max_iter 下（100），逻辑回归的梯度下降搜索没有收敛。

手动让 max_iter 更大即可。我在我的版本下测试，这样写就没有问题（让 max_iter = 1000）：

log_reg = LogisticRegression(max_iter=1000)

继续加油！：）