如何使用 sklearn 禁用 ConvergenceWarning？

Question

我正在使用 GridSearchCV 来优化 SVM 的超参数。我设置了最大迭代次数，因为我不能等待几个小时才能得到结果。我知道会有收敛警告。我只想忽略这些警告，不显示在终端中。

提前致谢。

原文由 Bohan Xu 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

追踪起来很痛苦，因为我看到的所有建议答案都行不通。最终对我有用的是示例代码 Early stopping of Stochastic Gradient Descent ：

 from sklearn.utils.testing import ignore_warnings
from sklearn.exceptions import ConvergenceWarning

然后你可以像这样注释一个函数：

 @ignore_warnings(category=ConvergenceWarning)
def my_function():
    # Code that triggers the warning

请注意，您无需直接从 warnings 导入任何内容。

我认为这很好，因为它只会在您需要的特定情况下而不是在全球范围内抑制警告。

原文由 simonharding 发布，翻译遵循 CC BY-SA 4.0 许可协议