逻辑回归调用中出现警告
来源: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)
继续加油!:)
00
相似问题