fit()什么时候停止在scikit中运行?

时间:2014-05-04 16:42:06

标签: python scikit-learn

我使用scikit-learn来训练分类器。我特别使用linear_model.LogisticRegression。但我的问题是:培训的停止标准是什么?!因为我没有看到任何表明时代数的参数!

随机森林也一样吗?

1 个答案:

答案 0 :(得分:3)

LogisticRegression的迭代次数没有硬性限制;相反,它尝试检测具有指定容差的收敛toltol越小,算法运行的时间越长。

从源代码中,我发现在训练开始之前,当目标梯度的范数小于其初始值的tol时,算法就会停止。这值得记录。

对于随机森林,当n_estimators树符合最大深度max_depth时,训练会停止,受参数min_samples_splitmin_samples_leafmax_leaf_nodes约束。树学习与迭代线性模型学习完全不同。