younggyoseo / vae-cf-pytorch

Variational Autoencoders for Collaborative Filtering - Implementation in PyTorch
130 stars 22 forks source link

fixed error on dataloading #4

Closed youngjae99 closed 3 years ago

youngjae99 commented 3 years ago

Hello, I tried to run your code since I'm studying how to implement a recommendation system using AutoEncoder. While I was running it, there was an error so I fixed it a bit and sent a Pull Request. I fixed some code in data.py file :) Thank you.

bupt-yxy commented 3 years ago

thanks you, It helps me a lot. thank you very much!!!

younggyoseo commented 3 years ago

Thanks! I'm not maintaining this code - but this will be helpful for others. --Younggyo