recommenders-team/recommenders

Multinomial VAE - performance

Open

#1.249 aberto em 25 de nov. de 2020

Ver no GitHub
 (1 comment) (0 reactions) (0 assignees)Python (2.972 forks)batch import
help wanted

Métricas do repositório

Stars
 (17.706 stars)
Métricas de merge de PR
 (Mesclagem média 6d 16h) (10 fundiu PRs em 30d)

Description

I've noticed some differences in training time and performance between this tensorflow 2 implementation and the original version:

  • lower ratings should be removed

user-to-movie interactions with rating <=3.5 are filtered out but they are used to generate test_data_te_ratings, val_data_te_ratings that are used to compute the metrics for the model. Could I ask why?

  • huge spike in memory consumption, using dataset 20m when the training is started will gobble up ~25gb RAM while the initial version will only use ~4gb RAM. Is this is a bug?

Thank you

Guia do colaborador