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

时间:2018-03-28 19:59:31

标签: tensorflow auc

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

  

权重:可选的Tensor,其等级为0或与标签相同的等级,并且必须可以向标签广播(即,所有维度必须为1,或与相应的标签维度相同)。

  

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

假设我想用重量测量两个AUC:一个用于男性,一个用于女性。

你能举例说明怎么做吗?

编辑:假设我有足够的课程,我不想将数据分成所有不同的类,而且我不想将足够的数据全部读入内存。也就是说,我想以流媒体方式进行。

1 个答案:

答案 0 :(得分:0)

假设您的标签是一个向量,将权重设置为一个向量,其中数据点属于一个行的行为0,而对于数据点不属于该类的行,为0将允许您计算该成员的AUC类。