我想请教一下为什么我用alphaFM跑出来的结果auc表现远不如LR

CastellanZhang / alphaFM

Multi-thread implementation of Factorization Machines with FTRL for binary-class classification problem.

MIT License

881 stars 275 forks source link

Closed fooSynaptic closed 4 years ago

fooSynaptic commented 4 years ago

在预测文件中，第一个是label1/-1，后面跟着的是概率，在计算auc的时候，我通过如下的方式来获取模型预测的概率

如上准备的概率和标签一起进入BinaryClassificationMetrics，最后表现的auc不超过1%，而LR接近90%，请问可能在哪里出现了问题？

caomeibaobao1994 commented 4 years ago

我理解，不管label是什么，模型预测为1的概率都是prob

fooSynaptic commented 4 years ago

i found fm version run correctly, closed.