多标签分类任务中sklearn—roc_auc_score的使用

sklearn:auc、roc_curve、roc_auc_score_NockinOnHeavensDoor的博客-CSDN博客

切记:

1。

train的时候,若出现valuerror,只出现一个class在y_true的情况时,可通过提高batchsize的大小解决,这种情况的发生是因为数据中一些数据是全1或者全0.

2。

再一个就是要明确函数里面的参数,是y_score,(根据模型的预测结果给的评分)而不是model的直接输出output。

正确:roc_auc_score(label,score)

你可能感兴趣的:(DL实验排坑记录,sklearn,分类,机器学习)