Python如何过滤特定的警告？

Question

如何在python中过滤特定模块的特定警告

MWE

ERROR:

cross_val_score(model, df_Xtrain,ytrain,cv=2,scoring='r2')

/usr/local/lib/python3.6/dist-packages/sklearn/linear_model/_ridge.py:148: LinAlgWarning: Ill-conditioned matrix (rcond=3.275e-20): result may not be accurate.
  overwrite_a=True).T

我的尝试

import warnings
warnings.filterwarnings(action='ignore',module='sklearn')

cross_val_score(model, df_Xtrain,ytrain,cv=2,scoring='r2')

This works but I assume it ignores all the warnings such as deprecation warnings. I would like to exclude only LinAlgWarning

要求像

import warnings
warnings.filterwarnings(action='ignore', category='LinAlgWarning', module='sklearn')

是否有类似于此过滤器特定警告的方法

Answer 1

---

过滤警告的更多python方法：

---

import logging
    
for name in logging.Logger.manager.loggerDict.keys():
    if ('LinAlgWarning' in name):
            logging.getLogger(name).setLevel(logging.CRITICAL)

#rest of the code starts here...

Answer 2

您必须将类别作为WarningClass而不是字符串传递：

from scipy.linalg import LinAlgWarning

warnings.filterwarnings(action='ignore', category=LinAlgWarning, module='sklearn')