Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Command for evaluating NeuralFM model #1

Open
rezacsedu opened this issue Sep 16, 2017 · 3 comments
Open

Command for evaluating NeuralFM model #1

rezacsedu opened this issue Sep 16, 2017 · 3 comments

Comments

@rezacsedu
Copy link

rezacsedu commented Sep 16, 2017

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
Copy link
Owner

hexiangnan commented Sep 16, 2017 via email

@rezacsedu
Copy link
Author

rezacsedu commented Sep 16, 2017

@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?

@hexiangnan
Copy link
Owner

hexiangnan commented Sep 17, 2017 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants