使用pytorch训练模型时,不停的报警告信息sklearn\utils\deprecation.py:143:FutureWarning解决方法

使用pytorch训练模型时,不停的报警告信息D:\anaconda\envs\pytorch\lib\site-packages\sklearn\utils\deprecation.py:143: FutureWarning解决方法

如下图所示的报警信息,总是不停的显示,十分的影响干扰我们观测训练的实时精度和结果!!!给出两种解决方案在这里插入图片描述
暴力解决
直接忽视该警告信息,在训练的过程中禁止输出该警告信息,解决方法如下:首先在代码中加入以下代码片段

import warnings
warnings.filterwarnings('ignore')

然后在训练时使用以下样式的命令

(pytorch) D:\reid\code>python -W ignore train.py

即可直接忽视警告信息,虽然感觉有点掩耳盗铃,但是不影响训练的结果

理性解决法
根据报警信息采取对应的解决方法:
如我遇到的该报警信息,意思就是sklearn.metrics.base模块在版本0.22中已经弃用,在版本0.24中将被删除。 所以应该从sklearn.metrics导入相应的类/函数。 现在无法从sklearn.metrics导入的所有内容都是私有API的一部分
也就是说该模块的使用方法以及变更了,找到对应的代码修改

#原来的模块引用方法
from sklearn.metrics.base import precision_recall_curve, auc

#修改
from sklearn.metrics import precision_recall_curve, auc

你可能感兴趣的:(代码)