请问在BERT多标签分类中如何设置阈值呢？

Open KevinChen1994 opened this issue 5 years ago • 0 comments

您好，感谢贡献代码，对我有很大的帮助！我在使用bert多多标签分类时，看到您的代码的eval中只计算了auc，代码在这里 current_auc, update_op_auc = tf.metrics.auc(label_id_, probabilities) 可以看到这里传入是label_id 和probabilities，比如在标签概率中我想设置阈值为0.5的话，该如何设置呢？多谢解答！

Apr 15 '20 10:04 KevinChen1994