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

Adding side features to train_mini_batch.py #11

Open
kuhanw opened this issue Jul 31, 2019 · 0 comments
Open

Adding side features to train_mini_batch.py #11

kuhanw opened this issue Jul 31, 2019 · 0 comments

Comments

@kuhanw
Copy link

kuhanw commented Jul 31, 2019

Hi,

Thank you for the code, I read the paper and I am interested in applying this model to a large scale bipartite graph with side features for each node. I was able to get "train_mini_batch.py" up and running on movielens 10 million. I am interested in modifying the code to incorporate side features.

Do you think this is feasible? I have been looking at "train.py" and as far as I can see, I should just be able to treat the side features as additional sparse matrices. In batch mode, I can look up the required rows of the side feature matrices analogous to train_u_indices_batch on line 284 of "train_mini_batch.py". Would I have to make any internal modifications to RecommenderSideInfoGAE?

Do you think this is feasible?

Thank You,

Kuhan

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

1 participant