逻辑回归调用中出现警告

来源:10-4 F1 Score

NLPLearner

2022-08-14

from sklearn.linear_model import LogisticRegression

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

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

写回答

1回答

liuyubobobo

2022-08-14

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


手动让 max_iter 更大即可。我在我的版本下测试,这样写就没有问题(让 max_iter = 1000):

log_reg = LogisticRegression(max_iter=1000)


继续加油!:)

0
0

Python3入门机器学习 经典算法与应用  

Python3+sklearn,兼顾原理、算法底层实现和框架使用。

5893 学习 · 2454 问题

查看课程