tensorflow-DeepFM icon indicating copy to clipboard operation
tensorflow-DeepFM copied to clipboard

Published 20 hours ago • verified publisher icon ChenglongChen

您好，请问如何修改为多分类，并输出给分类的概率

Open huminwudi opened this issue 5 years ago • 2 comments

huminwudi avatar Jul 26 '19 09:07 huminwudi

多分类你就把label换成one-hot格式，在代码里把self.out改成 tf.argmax，还有模型的输出维度也修改成对应的

DinLei avatar Aug 03 '19 16:08 DinLei

@huminwudi I would be interesting in a multi-category solution. Have you solved this issue?

aaad avatar Nov 04 '20 16:11 aaad