关于模型预测的infer接口

guoday / ctrNet-tool

This's the tool for CTR, including FM, FFM, NFFM and so on.

317 stars 93 forks source link

关于模型预测的infer接口 #2

Closed bestpredicts closed 5 years ago

bestpredicts commented 5 years ago

是不是可以只传入一个参数,比如 preds=model.infer(dev_data=(test[features])) 我看代码示例dev_data需要传入label? 还是说dev_data，必须传入label像这样 preds=model.infer(dev_data(test[features],test['HasDetections']))

guoday commented 5 years ago

实际上是不需要传label的。我也发现了这么问题，代码还没修改回来，后续会修改回来的。现在的代码不传label的话会报错