Final\tTrain=0.80606\tTest=0.370107
roc_auc_score is 0.82932847197120885
This is my libFM call:
./libFM -task c -method sgd -save_model model1 -learn_rate 0.01 -regular '0,0,0.01' -init_stdev 0.1 -train $train_file -test $test_file -iter 500 -dim '1,1,10' -out output.libfm
Is my concern valid and if yes, how do I go about fixing that gap?
Thanks!