如何在 tf.metrics.auc 中使用权重？

Question

tf.metrics.auc function in tensorflow 的文档说

权重：可选张量，其秩为 0，或与标签的秩相同，并且必须可广播到标签（即所有维度必须为 1，或与相应的标签维度相同）。

和

如果权重为无，权重默认为 1。使用权重 0 来屏蔽值。

假设我想使用权重来测量两个 AUC：一个用于男性，一个用于女性。

你能举例说明如何做到这一点吗？

编辑：假设我有足够的类，我不想将数据分成所有不同的类，并且有足够的数据，我不想将它们全部读入内存。也就是说，我想以流媒体方式进行。

Answer 1

假设您的标签是一个向量，将权重设置为一个向量，其中数据点属于某个类的行设置为 1，数据点不属于该类的行设置为 0，这样您就可以计算该类成员的 AUC类。