hexiangnan/neural_factorization_machine

Command for evaluating NeuralFM model

rezacsedu opened this issue · 3 comments

Hi there,

Could you please provide the command for evaluating NeuralFM model? -i.e. I was expecting something similar presented at https://github.com/hexiangnan/attentional_factorization_machine.

Best,

@hexiangnan, thanks so much.
I have seperated the test set from train/validation and did the evalaution seperately:

print("RMSE: ")
print(model.evaluate(data.Test_data)) # evaluate the results for an input set

BTW, could you please upload the script you used to convert the raw MovieLens (or frappe) dataset into LibFM format?