如何使所有sklearn警告静音

Question

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

  

“ UndefinedMetricWarning：F得分定义不正确，由于没有预测样本，因此将其设置为0.0”。

有人知道如何消除此警告吗？总的来说，我们可以让sklearn中的所有警告静音吗？

Answer 1

借助Python中的ignore模块，您可以轻松地warnings警告。

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))

这不会显示警告。