machine-learning - Why does xLearn fit function causes kernel crashes in Jupyter?

Question

I'm trying to make CTR (Click through rate) prediction using a python module named 'xlearn'. It enables me to implement a FFM (field-aware factorisation machine) quite easily.

However, I have a problem with the fit function ( supposed to train the model) which crashes the kernel of my jupyter notebook without any error messages.

Here is the code :

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

I've already tried to fit the model just after python ffm_model = xl.create_ffm() this also crashes the kernel without any error messages ...

Don't hesitate to share your ideas I'm really stuck here.

标签： machine-learningjupyter-notebookclassification

Answer 1

我没有意识到 xLearn 模块在终端中显示错误消息： Xlearn Imgae 错误消息