如何使所有 sklearn 警告静音

Question

我在 sklearn 中使用 F1_score 指标。对于某些训练数据集，y=1（罕见）集的总数为零，F1_score为零，这是正常的。但是 sklearn 给出了以下警告：

“UndefinedMetricWarning：F-score 定义不明确，由于没有预测样本而被设置为 0.0”。

有谁知道如何消除此警告？总的来说，我们可以在 sklearn 中消除各种警告吗？

Answer 1

您可以像这样在 Python 中借助 warnings 模块轻松 ignore 警告。

import warnings
warnings.filterwarnings('ignore') 

例如，

from sklearn.metrics import f1_score

yhat = [0] * 100
y = [0] * 90 + [1] * 10

print(f1_score(y, yhat))

这将引发警告。为了避免这种使用，

from sklearn.metrics import f1_score

import warnings
warnings.filterwarnings('ignore') 

yhat = [0] * 100
y = [0] * 90 + [1] * 10

print(f1_score(y, yhat))

这不会显示警告。