models
models copied to clipboard
Merlin Models is a collection of deep learning recommender system model reference implementations
### Description Currently none of our example notebooks do not have any code script showcasing how one can do model prediction locally, generate top-K recommendations from the prediction results. This...
Part of #669 ### Description Implement the save method on tensorflow models to write out both the tensorflow/keras artifacts as well as the merlin schema in a separate file.
### Description Currently, 05-Retrieval-Model.ipynb doesnt evaluate the model (model.evaluate). I think we should extend the example by model.evaluate. 1. We should add a description, when using a validation dataset in...
### Description Updating all example notebooks to have a customized NVIDIA logo
### Bug description Created a TwoTower Model or a Wide and Deep Model, firstly compile the model with the "adam" optiomizer and fit, and then call model.compile again with "sgd"...
### Bug description When we train Two-tower model and run model.evaluate() afterwards, we observe very different ndcg value wrt different batch size set in the `model.evaluate()`. if you remove `validation_data`...
Fixes #622 ### Goals :soccer: - [x] Decouple the top-k local prediction top-k evaluation from the retrieval contrastive learning task. - [ ] REfactor retrieval models and add a method...
### Bug description When training with (as best as I can tell) everything else held constant and the Dataset on the CPU, GPU memory consumption doesn't fluctuate during training. Throughout...
## Problem - Generating the top-k prediction from a trained the retrieval model currently requires the user to follow these steps (Not straightforward). Besides, the top-k recommender is a ModelBlock...