.pre-commit-config.yaml
CLA.md
CONTRIBUTING.md
LICENSE
MANIFEST.in
Makefile
README.md
pyproject.toml
pytest.ini
setup.cfg
setup.py
tox.ini
versioneer.py
conda/recipes/meta.yaml
merlin/datasets/__init__.py
merlin/datasets/synthetic.py
merlin/datasets/advertising/__init__.py
merlin/datasets/advertising/criteo/__init__.py
merlin/datasets/advertising/criteo/dataset.py
merlin/datasets/advertising/criteo/transformed/__init__.py
merlin/datasets/advertising/criteo/transformed/schema.pbtxt
merlin/datasets/ecommerce/__init__.py
merlin/datasets/ecommerce/aliccp/__init__.py
merlin/datasets/ecommerce/aliccp/dataset.py
merlin/datasets/ecommerce/aliccp/raw/__init__.py
merlin/datasets/ecommerce/aliccp/raw/schema.pbtxt
merlin/datasets/ecommerce/aliccp/transformed/__init__.py
merlin/datasets/ecommerce/aliccp/transformed/schema.pbtxt
merlin/datasets/ecommerce/booking/__init__.py
merlin/datasets/ecommerce/booking/dataset.py
merlin/datasets/ecommerce/booking/raw/__init__.py
merlin/datasets/ecommerce/booking/raw/schema.pbtxt
merlin/datasets/ecommerce/booking/transformed/__init__.py
merlin/datasets/ecommerce/booking/transformed/schema.pbtxt
merlin/datasets/ecommerce/dressipi/__init__.py
merlin/datasets/ecommerce/dressipi/dataset.py
merlin/datasets/ecommerce/dressipi/preprocessed/__init__.py
merlin/datasets/ecommerce/dressipi/preprocessed/schema.pbtxt
merlin/datasets/ecommerce/large/__init__.py
merlin/datasets/ecommerce/large/schema.json
merlin/datasets/ecommerce/sigir/__init__.py
merlin/datasets/ecommerce/sigir/browsing_train/__init__.py
merlin/datasets/ecommerce/sigir/browsing_train/schema.json
merlin/datasets/ecommerce/sigir/sku_information/__init__.py
merlin/datasets/ecommerce/sigir/sku_information/schema.json
merlin/datasets/ecommerce/small/__init__.py
merlin/datasets/ecommerce/small/schema.json
merlin/datasets/ecommerce/transactions/__init__.py
merlin/datasets/ecommerce/transactions/schema.pbtxt
merlin/datasets/entertainment/__init__.py
merlin/datasets/entertainment/movielens/__init__.py
merlin/datasets/entertainment/movielens/dataset.py
merlin/datasets/entertainment/movielens/100k/__init__.py
merlin/datasets/entertainment/movielens/100k/schema.json
merlin/datasets/entertainment/movielens/1m/__init__.py
merlin/datasets/entertainment/movielens/1m/schema.pbtxt
merlin/datasets/entertainment/movielens/1m-raw/__init__.py
merlin/datasets/entertainment/movielens/1m-raw/ratings/__init__.py
merlin/datasets/entertainment/movielens/1m-raw/ratings/schema.pbtxt
merlin/datasets/entertainment/movielens/25m/__init__.py
merlin/datasets/entertainment/movielens/25m/schema.pbtxt
merlin/datasets/entertainment/music_streaming/__init__.py
merlin/datasets/entertainment/music_streaming/schema.json
merlin/datasets/entertainment/tenrec_video/__init__.py
merlin/datasets/entertainment/tenrec_video/schema.pbtxt
merlin/datasets/social/__init__.py
merlin/datasets/social/schema.json
merlin/datasets/testing/__init__.py
merlin/datasets/testing/schema.json
merlin/datasets/testing/sequence_testing/__init__.py
merlin/datasets/testing/sequence_testing/schema.json
merlin/models/__init__.py
merlin/models/_version.py
merlin/models/api.py
merlin/models/io.py
merlin/models/py.typed
merlin/models/config/__init__.py
merlin/models/config/schema.py
merlin/models/implicit/__init__.py
merlin/models/lightfm/__init__.py
merlin/models/loader/__init__.py
merlin/models/loader/backend.py
merlin/models/tf/__init__.py
merlin/models/tf/loader.py
merlin/models/tf/typing.py
merlin/models/tf/blocks/__init__.py
merlin/models/tf/blocks/cross.py
merlin/models/tf/blocks/dlrm.py
merlin/models/tf/blocks/experts.py
merlin/models/tf/blocks/interaction.py
merlin/models/tf/blocks/mlp.py
merlin/models/tf/blocks/optimizer.py
merlin/models/tf/blocks/retrieval/__init__.py
merlin/models/tf/blocks/retrieval/base.py
merlin/models/tf/blocks/retrieval/matrix_factorization.py
merlin/models/tf/blocks/retrieval/two_tower.py
merlin/models/tf/blocks/sampling/__init__.py
merlin/models/tf/blocks/sampling/base.py
merlin/models/tf/blocks/sampling/cross_batch.py
merlin/models/tf/blocks/sampling/in_batch.py
merlin/models/tf/blocks/sampling/queue.py
merlin/models/tf/core/__init__.py
merlin/models/tf/core/aggregation.py
merlin/models/tf/core/base.py
merlin/models/tf/core/combinators.py
merlin/models/tf/core/encoder.py
merlin/models/tf/core/index.py
merlin/models/tf/core/prediction.py
merlin/models/tf/core/tabular.py
merlin/models/tf/distributed/__init__.py
merlin/models/tf/distributed/backend.py
merlin/models/tf/distributed/embedding.py
merlin/models/tf/experimental/sample_weight.py
merlin/models/tf/inputs/__init__.py
merlin/models/tf/inputs/base.py
merlin/models/tf/inputs/continuous.py
merlin/models/tf/inputs/embedding.py
merlin/models/tf/logging/callbacks.py
merlin/models/tf/losses/__init__.py
merlin/models/tf/losses/base.py
merlin/models/tf/losses/listwise.py
merlin/models/tf/losses/pairwise.py
merlin/models/tf/metrics/__init__.py
merlin/models/tf/metrics/evaluation.py
merlin/models/tf/metrics/topk.py
merlin/models/tf/models/__init__.py
merlin/models/tf/models/base.py
merlin/models/tf/models/benchmark.py
merlin/models/tf/models/ranking.py
merlin/models/tf/models/retrieval.py
merlin/models/tf/models/utils.py
merlin/models/tf/outputs/__init__.py
merlin/models/tf/outputs/base.py
merlin/models/tf/outputs/block.py
merlin/models/tf/outputs/classification.py
merlin/models/tf/outputs/contrastive.py
merlin/models/tf/outputs/regression.py
merlin/models/tf/outputs/topk.py
merlin/models/tf/outputs/sampling/__init__.py
merlin/models/tf/outputs/sampling/base.py
merlin/models/tf/outputs/sampling/in_batch.py
merlin/models/tf/outputs/sampling/popularity.py
merlin/models/tf/prediction_tasks/__init__.py
merlin/models/tf/prediction_tasks/base.py
merlin/models/tf/prediction_tasks/classification.py
merlin/models/tf/prediction_tasks/multi.py
merlin/models/tf/prediction_tasks/next_item.py
merlin/models/tf/prediction_tasks/regression.py
merlin/models/tf/prediction_tasks/retrieval.py
merlin/models/tf/transformers/__init__.py
merlin/models/tf/transformers/block.py
merlin/models/tf/transformers/transforms.py
merlin/models/tf/transforms/__init__.py
merlin/models/tf/transforms/bias.py
merlin/models/tf/transforms/features.py
merlin/models/tf/transforms/negative_sampling.py
merlin/models/tf/transforms/noise.py
merlin/models/tf/transforms/regularization.py
merlin/models/tf/transforms/sequence.py
merlin/models/tf/transforms/tensor.py
merlin/models/tf/utils/__init__.py
merlin/models/tf/utils/batch_utils.py
merlin/models/tf/utils/repr_utils.py
merlin/models/tf/utils/search_utils.py
merlin/models/tf/utils/testing_utils.py
merlin/models/tf/utils/tf_utils.py
merlin/models/torch/__init__.py
merlin/models/torch/batch.py
merlin/models/torch/block.py
merlin/models/torch/container.py
merlin/models/torch/functional.py
merlin/models/torch/predict.py
merlin/models/torch/registry.py
merlin/models/torch/router.py
merlin/models/torch/schema.py
merlin/models/torch/blocks/__init__.py
merlin/models/torch/blocks/attention.py
merlin/models/torch/blocks/cross.py
merlin/models/torch/blocks/dlrm.py
merlin/models/torch/blocks/experts.py
merlin/models/torch/blocks/mlp.py
merlin/models/torch/inputs/__init__.py
merlin/models/torch/inputs/embedding.py
merlin/models/torch/inputs/select.py
merlin/models/torch/inputs/tabular.py
merlin/models/torch/models/__init__.py
merlin/models/torch/models/base.py
merlin/models/torch/models/ranking.py
merlin/models/torch/outputs/__init__.py
merlin/models/torch/outputs/base.py
merlin/models/torch/outputs/classification.py
merlin/models/torch/outputs/contrastive.py
merlin/models/torch/outputs/regression.py
merlin/models/torch/outputs/tabular.py
merlin/models/torch/outputs/sampling/__init__.py
merlin/models/torch/outputs/sampling/in_batch.py
merlin/models/torch/outputs/sampling/popularity.py
merlin/models/torch/transforms/__init__.py
merlin/models/torch/transforms/agg.py
merlin/models/torch/transforms/bias.py
merlin/models/torch/transforms/sequences.py
merlin/models/torch/transforms/tuple.py
merlin/models/torch/utils/__init__.py
merlin/models/torch/utils/module_utils.py
merlin/models/torch/utils/torchscript_utils.py
merlin/models/torch/utils/traversal_utils.py
merlin/models/utils/__init__.py
merlin/models/utils/ci_utils.py
merlin/models/utils/constants.py
merlin/models/utils/dataset.py
merlin/models/utils/dependencies.py
merlin/models/utils/doc_utils.py
merlin/models/utils/example_utils.py
merlin/models/utils/misc_utils.py
merlin/models/utils/nvt_utils.py
merlin/models/utils/registry.py
merlin/models/utils/schema_utils.py
merlin/models/xgb/__init__.py
merlin_models.egg-info/PKG-INFO
merlin_models.egg-info/SOURCES.txt
merlin_models.egg-info/dependency_links.txt
merlin_models.egg-info/requires.txt
merlin_models.egg-info/top_level.txt
requirements/base.txt
requirements/dev.txt
requirements/docs.txt
requirements/horovod-cpu-environment.yml
requirements/horovod.txt
requirements/implicit.txt
requirements/lightfm.txt
requirements/nvtabular.txt
requirements/pytorch.txt
requirements/tensorflow.txt
requirements/test.txt
requirements/transformers.txt
requirements/xgboost.txt
tests/__init__.py
tests/conftest.py
tests/common/__init__.py
tests/common/tf/__init__.py
tests/common/tf/tests_utils.py
tests/common/tf/retrieval/__init__.py
tests/common/tf/retrieval/retrieval_config.py
tests/common/tf/retrieval/retrieval_tests_common.py
tests/common/tf/retrieval/retrieval_utils.py
tests/integration/__init__.py
tests/integration/tf/__init__.py
tests/integration/tf/test_ci_01_getting_started.py
tests/integration/tf/test_ci_02_dataschema.py
tests/integration/tf/test_ci_03_exploring_different_models.py
tests/integration/tf/test_ci_04_export_ranking_models.py
tests/integration/tf/test_ci_05_export_retrieval_model.py
tests/integration/tf/test_ci_06_advanced_own_architecture.py
tests/integration/tf/retrieval/__init__.py
tests/integration/tf/retrieval/test_integration_retrieval.py
tests/integration/torch/__init__.py
tests/integration/torch/test_multi_gpu.py
tests/unit/__init__.py
tests/unit/config/test_schema.py
tests/unit/datasets/__init__.py
tests/unit/datasets/test_advertising.py
tests/unit/datasets/test_ecommerce.py
tests/unit/datasets/test_entertainment.py
tests/unit/datasets/test_social.py
tests/unit/datasets/test_synthetic.py
tests/unit/implicit/__init__.py
tests/unit/implicit/test_implicit.py
tests/unit/lightfm/__init__.py
tests/unit/lightfm/test_lightfm.py
tests/unit/tf/__init__.py
tests/unit/tf/_conftest.py
tests/unit/tf/test_core.py
tests/unit/tf/test_loader.py
tests/unit/tf/test_public_api.py
tests/unit/tf/blocks/__init__.py
tests/unit/tf/blocks/test_cross.py
tests/unit/tf/blocks/test_dlrm.py
tests/unit/tf/blocks/test_interactions.py
tests/unit/tf/blocks/test_mlp.py
tests/unit/tf/blocks/test_optimizer.py
tests/unit/tf/blocks/retrieval/__init__.py
tests/unit/tf/blocks/retrieval/test_base.py
tests/unit/tf/blocks/retrieval/test_matrix_factorization.py
tests/unit/tf/blocks/retrieval/test_two_tower.py
tests/unit/tf/blocks/sampling/__init__.py
tests/unit/tf/blocks/sampling/test_cross_batch.py
tests/unit/tf/blocks/sampling/test_in_batch.py
tests/unit/tf/core/__init__.py
tests/unit/tf/core/test_aggregation.py
tests/unit/tf/core/test_base.py
tests/unit/tf/core/test_combinators.py
tests/unit/tf/core/test_encoder.py
tests/unit/tf/core/test_index.py
tests/unit/tf/core/test_prediction.py
tests/unit/tf/core/test_tabular.py
tests/unit/tf/examples/__init__.py
tests/unit/tf/examples/test_01_getting_started.py
tests/unit/tf/examples/test_02_dataschema.py
tests/unit/tf/examples/test_03_exploring_different_models.py
tests/unit/tf/examples/test_04_export_ranking_models.py
tests/unit/tf/examples/test_05_export_retrieval_model.py
tests/unit/tf/examples/test_06_advanced_own_architecture.py
tests/unit/tf/examples/test_07_train_traditional_models.py
tests/unit/tf/examples/test_08_session_based_next_item_prediction.py
tests/unit/tf/examples/test_usecase_accelerate_training_by_lazyadam.py
tests/unit/tf/examples/test_usecase_data_parallel.py
tests/unit/tf/examples/test_usecase_incremental_training_layer_freezing.py
tests/unit/tf/examples/test_usecase_pretrained_embeddings.py
tests/unit/tf/examples/test_usecase_ranking_with_multitask_learning.py
tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py
tests/unit/tf/experimental/__init__.py
tests/unit/tf/experimental/test_sample_weight.py
tests/unit/tf/horovod/__init__.py
tests/unit/tf/horovod/test_embedding.py
tests/unit/tf/horovod/test_horovod.py
tests/unit/tf/inputs/__init__.py
tests/unit/tf/inputs/test_base.py
tests/unit/tf/inputs/test_block.py
tests/unit/tf/inputs/test_continuous.py
tests/unit/tf/inputs/test_embedding.py
tests/unit/tf/inputs/test_tabular.py
tests/unit/tf/layers/__init__.py
tests/unit/tf/layers/test_queue.py
tests/unit/tf/losses/__init__.py
tests/unit/tf/losses/test_losses.py
tests/unit/tf/metrics/__init__.py
tests/unit/tf/metrics/test_metrics.py
tests/unit/tf/metrics/test_metrics_topk.py
tests/unit/tf/models/__init__.py
tests/unit/tf/models/test_base.py
tests/unit/tf/models/test_benchmark.py
tests/unit/tf/models/test_ranking.py
tests/unit/tf/models/test_retrieval.py
tests/unit/tf/outputs/__init__.py
tests/unit/tf/outputs/test_base.py
tests/unit/tf/outputs/test_block.py
tests/unit/tf/outputs/test_classification.py
tests/unit/tf/outputs/test_contrastive.py
tests/unit/tf/outputs/test_regression.py
tests/unit/tf/outputs/test_sampling.py
tests/unit/tf/outputs/test_topk.py
tests/unit/tf/prediction_tasks/__init__.py
tests/unit/tf/prediction_tasks/test_classification.py
tests/unit/tf/prediction_tasks/test_multi_task.py
tests/unit/tf/prediction_tasks/test_next_item.py
tests/unit/tf/prediction_tasks/test_regression.py
tests/unit/tf/prediction_tasks/test_retrieval.py
tests/unit/tf/prediction_tasks/test_sampling.py
tests/unit/tf/transformers/__init__.py
tests/unit/tf/transformers/test_block.py
tests/unit/tf/transformers/test_transforms.py
tests/unit/tf/transforms/__init__.py
tests/unit/tf/transforms/test_bias.py
tests/unit/tf/transforms/test_features.py
tests/unit/tf/transforms/test_negative_sampling.py
tests/unit/tf/transforms/test_noise.py
tests/unit/tf/transforms/test_sequence.py
tests/unit/tf/transforms/test_tensor.py
tests/unit/tf/utils/__init__.py
tests/unit/tf/utils/test_batch.py
tests/unit/tf/utils/test_dataset.py
tests/unit/tf/utils/test_tf_utils.py
tests/unit/torch/__init__.py
tests/unit/torch/_conftest.py
tests/unit/torch/test_batch.py
tests/unit/torch/test_block.py
tests/unit/torch/test_container.py
tests/unit/torch/test_functional.py
tests/unit/torch/test_predict.py
tests/unit/torch/test_router.py
tests/unit/torch/test_schema.py
tests/unit/torch/blocks/__init__.py
tests/unit/torch/blocks/test_attention.py
tests/unit/torch/blocks/test_cross.py
tests/unit/torch/blocks/test_dlrm.py
tests/unit/torch/blocks/test_experts.py
tests/unit/torch/blocks/test_mlp.py
tests/unit/torch/inputs/__init__.py
tests/unit/torch/inputs/test_embedding.py
tests/unit/torch/inputs/test_select.py
tests/unit/torch/inputs/test_tabular.py
tests/unit/torch/models/__init__.py
tests/unit/torch/models/test_base.py
tests/unit/torch/models/test_ranking.py
tests/unit/torch/outputs/__init__.py
tests/unit/torch/outputs/test_base.py
tests/unit/torch/outputs/test_classification.py
tests/unit/torch/outputs/test_constrastive.py
tests/unit/torch/outputs/test_regression.py
tests/unit/torch/outputs/test_tabular.py
tests/unit/torch/outputs/sampling/__init__.py
tests/unit/torch/outputs/sampling/test_in_batch.py
tests/unit/torch/outputs/sampling/test_popularity.py
tests/unit/torch/transforms/__init__.py
tests/unit/torch/transforms/test_agg.py
tests/unit/torch/transforms/test_bias.py
tests/unit/torch/transforms/test_sequences.py
tests/unit/torch/transforms/test_tuple.py
tests/unit/torch/utils/__init__.py
tests/unit/torch/utils/test_module_utils.py
tests/unit/torch/utils/test_torchscript_utils.py
tests/unit/torch/utils/test_traversal_utils.py
tests/unit/utils/test_schema_utils.py
tests/unit/xgb/__init__.py
tests/unit/xgb/test_xgboost.py