models
Merlin Models is a collection of deep learning recommender system model reference implementations
File Explorer
Download Latest Version (.zip)- --new-model-addition.md
- bug-report.md
- documentation-request.md
- feature-request.md
- submit-question.md
- task.md
- branches.yaml
- check-base-branch.yaml
- cpu-horovod.yml
- cpu-nvtabular.yml
- cpu-systems.yml
- cpu-t4r.yml
- datasets.yml
- docs-build.yaml
- docs-preview-pr.yaml
- docs-remove-stale-reviews.yaml
- docs-sched-rebuild.yaml
- gpu-multi.yml
- gpu.yml
- implicit.yml
- lightfm.yml
- lint.yml
- packages.yaml
- pytorch.yml
- release-drafter.yaml
- require-label.yaml
- tensorflow.yml
- triage.yml
- xgboost.yml
- actionlint-matcher.json
- actionlint.yaml
- copy-pr-bot.yaml
- PULL_REQUEST_TEMPLATE.md
- release-drafter.yml
- get_changed_backends.py
- ignore_codespell_words.txt
- lint_tensorflow.sh
- pr.gpu.Jenkinsfile
- test_integration.sh
- test_unit.sh
- meta.yaml
- dcn-v2.png
- dlrm.png
- mf.png
- mmoe.png
- ple.png
- two-tower.png
- youtube-dnn.png
- custom.css
- versions.css
- rtd-version-switcher.js
- favicon.png
- NVIDIA-LogoBlack.svg
- NVIDIA-LogoWhite.svg
- footer.html
- layout.html
- merlin-ecosystem.html
- versions.html
- additional_resources.rst
- api.rst
- conf.py
- index.rst
- models_overview.md
- toc.yaml
- Makefile
- README.md
- bi-lstm_ecommerce.png
- DCN.png
- DLRM.png
- experiments.png
- merlin.png
- mlp.png
- mlp_ecommerce.png
- mtl_architectures.png
- ncf.png
- residual_interaction.png
- schema.png
- TwoTower.png
- wide_and_deep.png
- XLNet_ecommerce.png
- hvd_wrapper.sh
- install_sparse_operation_kit.sh
- __init__.py
- accelerate-training-of-large-embedding-tables-by-LazyAdam.ipynb
- entertainment-with-pretrained-embeddings.ipynb
- incremental-training-with-layer-freezing.ipynb
- multi-gpu-data-parallel-training.ipynb
- ranking_with_multitask_learning.ipynb
- retrieval-with-hyperparameter-optimization.ipynb
- 01-Getting-started.ipynb
- 02-Merlin-Models-and-NVTabular-integration.ipynb
- 03-Exploring-different-models.ipynb
- 04-Exporting-ranking-models.ipynb
- 05-Retrieval-Model.ipynb
- 06-Define-your-own-architecture-with-Merlin-Models.ipynb
- 07-Train-traditional-ML-models-using-the-Merlin-Models-API.ipynb
- 08-Train-a-model-for-session-based-next-item-prediction.ipynb
- __init__.py
- README.md
- __init__.py
- schema.pbtxt
- __init__.py
- dataset.py
- __init__.py
- __init__.py
- schema.pbtxt
- __init__.py
- schema.pbtxt
- __init__.py
- dataset.py
- __init__.py
- schema.pbtxt
- __init__.py
- schema.pbtxt
- __init__.py
- dataset.py
- __init__.py
- schema.pbtxt
- __init__.py
- dataset.py
- __init__.py
- schema.json
- __init__.py
- schema.json
- __init__.py
- schema.json
- __init__.py
- __init__.py
- schema.json
- __init__.py
- schema.pbtxt
- __init__.py
- __init__.py
- schema.json
- __init__.py
- schema.pbtxt
- __init__.py
- schema.pbtxt
- __init__.py
- __init__.py
- schema.pbtxt
- __init__.py
- dataset.py
- __init__.py
- schema.json
- __init__.py
- schema.pbtxt
- __init__.py
- __init__.py
- schema.json
- __init__.py
- schema.json
- __init__.py
- schema.json
- __init__.py
- synthetic.py
- __init__.py
- schema.py
- __init__.py
- __init__.py
- __init__.py
- backend.py
- __init__.py
- base.py
- matrix_factorization.py
- two_tower.py
- __init__.py
- base.py
- cross_batch.py
- in_batch.py
- queue.py
- __init__.py
- cross.py
- dlrm.py
- experts.py
- interaction.py
- mlp.py
- optimizer.py
- __init__.py
- aggregation.py
- base.py
- combinators.py
- encoder.py
- index.py
- prediction.py
- tabular.py
- __init__.py
- backend.py
- embedding.py
- sample_weight.py
- __init__.py
- base.py
- continuous.py
- embedding.py
- callbacks.py
- __init__.py
- base.py
- listwise.py
- pairwise.py
- __init__.py
- evaluation.py
- topk.py
- __init__.py
- base.py
- benchmark.py
- ranking.py
- retrieval.py
- utils.py
- __init__.py
- base.py
- in_batch.py
- popularity.py
- __init__.py
- base.py
- block.py
- classification.py
- contrastive.py
- regression.py
- topk.py
- __init__.py
- base.py
- classification.py
- multi.py
- next_item.py
- regression.py
- retrieval.py
- __init__.py
- block.py
- transforms.py
- __init__.py
- bias.py
- features.py
- negative_sampling.py
- noise.py
- regularization.py
- sequence.py
- tensor.py
- __init__.py
- batch_utils.py
- repr_utils.py
- search_utils.py
- testing_utils.py
- tf_utils.py
- __init__.py
- loader.py
- typing.py
- __init__.py
- attention.py
- cross.py
- dlrm.py
- experts.py
- mlp.py
- __init__.py
- embedding.py
- select.py
- tabular.py
- __init__.py
- base.py
- ranking.py
- __init__.py
- in_batch.py
- popularity.py
- __init__.py
- base.py
- classification.py
- contrastive.py
- regression.py
- tabular.py
- __init__.py
- agg.py
- bias.py
- sequences.py
- tuple.py
- __init__.py
- module_utils.py
- torchscript_utils.py
- traversal_utils.py
- __init__.py
- batch.py
- block.py
- container.py
- functional.py
- predict.py
- registry.py
- router.py
- schema.py
- __init__.py
- ci_utils.py
- constants.py
- dataset.py
- dependencies.py
- doc_utils.py
- example_utils.py
- misc_utils.py
- nvt_utils.py
- registry.py
- schema_utils.py
- __init__.py
- __init__.py
- _version.py
- api.py
- io.py
- py.typed
- base.txt
- dev.txt
- docs.txt
- horovod-cpu-environment.yml
- horovod.txt
- implicit.txt
- lightfm.txt
- nvtabular.txt
- pytorch.txt
- tensorflow.txt
- test.txt
- transformers.txt
- xgboost.txt
- __init__.py
- retrieval_config.py
- retrieval_tests_common.py
- retrieval_utils.py
- __init__.py
- tests_utils.py
- __init__.py
- __init__.py
- test_integration_retrieval.py
- __init__.py
- test_ci_01_getting_started.py
- test_ci_02_dataschema.py
- test_ci_03_exploring_different_models.py
- test_ci_04_export_ranking_models.py
- test_ci_05_export_retrieval_model.py
- test_ci_06_advanced_own_architecture.py
- __init__.py
- test_multi_gpu.py
- __init__.py
- test_schema.py
- __init__.py
- test_advertising.py
- test_ecommerce.py
- test_entertainment.py
- test_social.py
- test_synthetic.py
- __init__.py
- test_implicit.py
- __init__.py
- test_lightfm.py
- __init__.py
- test_base.py
- test_matrix_factorization.py
- test_two_tower.py
- __init__.py
- test_cross_batch.py
- test_in_batch.py
- __init__.py
- test_cross.py
- test_dlrm.py
- test_interactions.py
- test_mlp.py
- test_optimizer.py
- __init__.py
- test_aggregation.py
- test_base.py
- test_combinators.py
- test_encoder.py
- test_index.py
- test_prediction.py
- test_tabular.py
- __init__.py
- test_01_getting_started.py
- test_02_dataschema.py
- test_03_exploring_different_models.py
- test_04_export_ranking_models.py
- test_05_export_retrieval_model.py
- test_06_advanced_own_architecture.py
- test_07_train_traditional_models.py
- test_08_session_based_next_item_prediction.py
- test_usecase_accelerate_training_by_lazyadam.py
- test_usecase_data_parallel.py
- test_usecase_incremental_training_layer_freezing.py
- test_usecase_pretrained_embeddings.py
- test_usecase_ranking_with_multitask_learning.py
- test_usecase_retrieval_with_hpo.py
- __init__.py
- test_sample_weight.py
- __init__.py
- test_embedding.py
- test_horovod.py
- __init__.py
- test_base.py
- test_block.py
- test_continuous.py
- test_embedding.py
- test_tabular.py
- __init__.py
- test_queue.py
- __init__.py
- test_losses.py
- __init__.py
- test_metrics.py
- test_metrics_topk.py
- __init__.py
- test_base.py
- test_benchmark.py
- test_ranking.py
- test_retrieval.py
- __init__.py
- test_base.py
- test_block.py
- test_classification.py
- test_contrastive.py
- test_regression.py
- test_sampling.py
- test_topk.py
- __init__.py
- test_classification.py
- test_multi_task.py
- test_next_item.py
- test_regression.py
- test_retrieval.py
- test_sampling.py
- __init__.py
- test_block.py
- test_transforms.py
- __init__.py
- test_bias.py
- test_features.py
- test_negative_sampling.py
- test_noise.py
- test_sequence.py
- test_tensor.py
- __init__.py
- test_batch.py
- test_dataset.py
- test_tf_utils.py
- __init__.py
- _conftest.py
- test_core.py
- test_loader.py
- test_public_api.py
- __init__.py
- test_attention.py
- test_cross.py
- test_dlrm.py
- test_experts.py
- test_mlp.py
- __init__.py
- test_embedding.py
- test_select.py
- test_tabular.py
- __init__.py
- test_base.py
- test_ranking.py
- __init__.py
- test_in_batch.py
- test_popularity.py
- __init__.py
- test_base.py
- test_classification.py
- test_constrastive.py
- test_regression.py
- test_tabular.py
- __init__.py
- test_agg.py
- test_bias.py
- test_sequences.py
- test_tuple.py
- __init__.py
- test_module_utils.py
- test_torchscript_utils.py
- test_traversal_utils.py
- __init__.py
- _conftest.py
- test_batch.py
- test_block.py
- test_container.py
- test_functional.py
- test_predict.py
- test_router.py
- test_schema.py
- test_schema_utils.py
- __init__.py
- test_xgboost.py
- __init__.py
- __init__.py
- conftest.py
- .gitignore
- .pre-commit-config.yaml
- CLA.md
- CONTRIBUTING.md
- LICENSE
- Makefile
- MANIFEST.in
- pyproject.toml
- pytest.ini
- README.md
- setup.cfg
- setup.py
- tox.ini
- versioneer.py
// repository documentation
Was this content helpful?
(0 ratings)