分类：AUC

2021
07-01

Pytorch训练模型得到输出后计算F1-Score 和AUC的操作

1、计算F1-Score对于二分类来说，假设batchsize大小为64的话，那么模型一个batch的输出应该是torch.size([64,2])，所以首先做的是得到这个二维矩阵的每一行的最大索引值，然后添加到一个列表中，同时把标签也添加到一个列表中，最后使用sklearn中计算F1的工具包进行计算，代码如下importnumpyasnpimportsklearn.metricsimportf1_scoreprob_all=[]lable_all=[]fori,(data,label)intqdm(train_data_loader):prob... 继续阅读 >

2020
10-10

python计算auc的方法

1、安装scikit-learn1.1Scikit-learn依赖Python(>=2.6or>=3.3),NumPy(>=1.6.1),SciPy(>=0.9).分别查看上述三个依赖的版本：python -V 结果：Python 2.7.3python -c 'import scipy; print scipy.version.version'scipy版本结果：0.9.0python -c "import numpy; print numpy.version.version"numpy结果：1.10.21.2Scikit-learn安装如果你已经... 继续阅读 >

2020
10-08

keras用auc做metrics以及早停实例

我就废话不多说了，大家还是直接看代码吧~importtensorflowastffromsklearn.metricsimportroc_auc_scoredefauroc(y_true,y_pred):returntf.py_func(roc_auc_score,(y_true,y_pred),tf.double)#BuildModel...model.compile(loss='categorical_crossentropy',optimizer='adam',metrics=['accuracy',auroc])完整例子：defauc(y_true,y_pred):auc=tf.metrics.auc(y_true,y_pred)[1]K.get_session().run(tf.loc... 继续阅读 >