【发布时间】:2019-06-26 08:01:03
【问题描述】:
我正在尝试使用名为“xlearn”的 python 模块进行 CTR(点击率)预测。 它使我能够很容易地实现 FFM(场感知分解机)。
但是,我遇到了 fit 函数(应该训练模型)的问题,它使我的 jupyter notebook 的内核崩溃而没有任何错误消息。
代码如下:
import xlearn as xl
ffm_model = xl.create_ffm()
param = {'task':'binary', 'lr':0.2, 'lambda':0.002, 'metric':'acc'}
ffm_model.setTrain('ffm_train.txt')
ffm_model.fit(param, "./model.out") #this line crashes the kernel
我已经尝试在 python ffm_model = xl.create_ffm() 之后拟合模型,这也会导致内核崩溃而没有任何错误消息...
不要犹豫,分享你的想法我真的被困在这里了。
【问题讨论】:
标签: machine-learning jupyter-notebook classification