使用 sklearn 进行逻辑回归中的学习率

数据挖掘 机器学习 scikit-学习 逻辑回归
2021-10-14 10:19:22

在sklearn中,对于逻辑回归,可以定义惩罚、正则化率等变量。有没有办法设置学习率?

2个回答

sklearn.linear_model.LogisticRegression不使用 SGD,所以没有学习率。

我认为sklearn.linear_model.SGDClassifier这就是你所需要的,它是一个带有 SGD 训练的线性分类器。

参考

http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html

http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.SGDClassifier.html

根据sklearn的Logistic源码,用来最小化损失函数的求解器是SAG求解器(Stochastic Average Gradient)。本文定义了这种方法,在这个链接中有下垂求解器的实现。求解器的这个实现使用了一种方法来获得步长(学习率),所以没有办法可以改变学习率(除非你想改变源代码)。