使用DNNLinearCombinedClassifier进行多标签分类

时间:2019-01-23 14:43:08

标签: python tensorflow tensorflow-estimator

我想针对多标签分类问题运行DNNLinearCombinedClassifier(“宽和深”基础教程),即每个样本应分类为: 1.“单击”或“未单击” 2.“隐藏”或“未隐藏”

据我了解,DNNLinearCombinedClassifier的n_classes参数仅适用于多类标签(即每个样本可以是N个类之一)。

tf.contrib.estimator.multi_label_head不能与tf.estimator.DNNLinearCombinedClassifier一起使用,后者在构造函数中缺少head参数。

我想保留两个不同的标签,而不是将问题转换为 能够监控精度和召回措施的4类问题。

0 个答案:

没有答案