├── .github ├── pull_request_template.md └── workflows │ ├── ci.yml │ ├── doc.yml │ └── publish-to-test-pypi.yml ├── .gitignore ├── .pylintrc ├── LICENSE ├── Makefile ├── check_version.py ├── deepr ├── __init__.py ├── cli │ ├── __init__.py │ └── main.py ├── config │ ├── __init__.py │ ├── base.py │ ├── experimental.py │ ├── macros.py │ └── references.py ├── examples │ ├── __init__.py │ ├── movielens │ │ ├── __init__.py │ │ ├── configs │ │ │ ├── README.md │ │ │ ├── average │ │ │ │ ├── config.jsonnet │ │ │ │ └── macros.jsonnet │ │ │ ├── average_svd │ │ │ │ ├── config.jsonnet │ │ │ │ └── macros.jsonnet │ │ │ ├── common │ │ │ │ ├── end.jsonnet │ │ │ │ ├── evaluate.jsonnet │ │ │ │ ├── macros.jsonnet │ │ │ │ ├── start.jsonnet │ │ │ │ └── train.jsonnet │ │ │ ├── vae │ │ │ │ ├── config.jsonnet │ │ │ │ └── macros.jsonnet │ │ │ └── vae_svd │ │ │ │ ├── config.jsonnet │ │ │ │ └── macros.jsonnet │ │ ├── jobs │ │ │ ├── __init__.py │ │ │ ├── build_records.py │ │ │ ├── evaluate.py │ │ │ ├── init.py │ │ │ ├── predict.py │ │ │ └── svd.py │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── average.py │ │ │ ├── bpr.py │ │ │ ├── losses.py │ │ │ ├── multi.py │ │ │ ├── ns.py │ │ │ ├── transformer.py │ │ │ └── vae.py │ │ ├── macros │ │ │ ├── __init__.py │ │ │ └── paths.py │ │ ├── metrics │ │ │ ├── __init__.py │ │ │ ├── ndcg.py │ │ │ └── recall.py │ │ ├── prepros │ │ │ ├── __init__.py │ │ │ ├── csv.py │ │ │ └── record.py │ │ ├── readers │ │ │ ├── __init__.py │ │ │ └── csv.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ └── fields.py │ └── multiply │ │ ├── __init__.py │ │ ├── configs │ │ ├── config.json │ │ └── macros.json │ │ ├── jobs │ │ ├── __init__.py │ │ ├── build.py │ │ └── predict.py │ │ ├── layers │ │ ├── __init__.py │ │ ├── loss.py │ │ └── model.py │ │ ├── macros │ │ ├── __init__.py │ │ └── paths.py │ │ └── prepros │ │ ├── __init__.py │ │ ├── default.py │ │ └── inference.py ├── exporters │ ├── __init__.py │ ├── base.py │ ├── best_checkpoint.py │ ├── save_variables.py │ └── saved_model.py ├── hooks │ ├── __init__.py │ ├── base.py │ ├── early_stopping.py │ ├── log_variables_init.py │ ├── logging_tensor.py │ ├── num_params.py │ ├── steps_per_sec.py │ └── summary.py ├── initializers │ ├── __init__.py │ └── checkpoint_initializer.py ├── io │ ├── __init__.py │ ├── hdfs.py │ ├── json.py │ ├── parquet.py │ └── path.py ├── jobs │ ├── __init__.py │ ├── base.py │ ├── cleanup_checkpoints.py │ ├── combinators.py │ ├── copy_dir.py │ ├── evaluate.py │ ├── export_xla_model_metadata.py │ ├── log_metric.py │ ├── mlflow_save_configs.py │ ├── mlflow_save_info.py │ ├── optimize_saved_model.py │ ├── params_tuner.py │ ├── save_dataset.py │ ├── trainer.py │ ├── trainer_base.py │ ├── trainer_keras.py │ ├── yarn_config.py │ ├── yarn_launcher.py │ └── yarn_trainer.py ├── layers │ ├── __init__.py │ ├── base.py │ ├── bpr.py │ ├── bpr_max.py │ ├── click_rank.py │ ├── combinators.py │ ├── core.py │ ├── dropout.py │ ├── embedding.py │ ├── lookup.py │ ├── lstm.py │ ├── mask.py │ ├── multi.py │ ├── multi_css.py │ ├── nce_loss.py │ ├── reduce.py │ ├── size.py │ ├── slice.py │ ├── sparse.py │ ├── string.py │ ├── top_one.py │ ├── top_one_max.py │ ├── transformer.py │ └── triplet_precision.py ├── macros │ ├── __init__.py │ ├── dynamic.py │ ├── graphite_init.py │ └── mlflow_init.py ├── metrics │ ├── __init__.py │ ├── accuracy.py │ ├── base.py │ ├── core.py │ ├── mean.py │ ├── step.py │ └── variable.py ├── optimizers │ ├── __init__.py │ ├── base.py │ └── core.py ├── predictors │ ├── __init__.py │ ├── base.py │ ├── proto.py │ └── saved_model.py ├── prepros │ ├── __init__.py │ ├── base.py │ ├── combinators.py │ ├── core.py │ ├── lookup.py │ └── record.py ├── readers │ ├── __init__.py │ ├── base.py │ ├── generator.py │ └── record.py ├── utils │ ├── __init__.py │ ├── broadcasting.py │ ├── checkpoint.py │ ├── datastruct.py │ ├── exceptions.py │ ├── field.py │ ├── graph.py │ ├── graphite.py │ ├── iter.py │ ├── mlflow.py │ ├── tables.py │ ├── tf2xla_pb2.py │ └── uuid.py ├── version.py ├── vocab │ ├── __init__.py │ └── base.py └── writers │ ├── __init__.py │ ├── base.py │ └── record.py ├── docs ├── .gitignore ├── API │ └── core.rst ├── CHANGELOG.rst ├── CONTRIBUTING.rst ├── Makefile ├── README.rst ├── conf.py ├── getting_started │ ├── advanced.ipynb │ ├── config.ipynb │ ├── pipeline.ipynb │ ├── quickstart.ipynb │ └── tuning.ipynb ├── index.rst └── movielens │ └── movielens.ipynb ├── mypy.ini ├── requirements-gpu.txt ├── requirements-test.txt ├── requirements.txt ├── setup.py └── tests ├── conftest.py ├── integration └── examples │ └── multiply │ ├── test_example_multiply_configs.py │ └── test_example_multiply_pipeline.py └── unit ├── cli └── test_cli_main.py ├── config ├── test_config_base.py ├── test_config_experimental.py ├── test_config_macros.py └── test_config_references.py ├── exporters ├── test_exporters_best_checkpoint.py └── test_exporters_save_variables.py ├── io ├── test_io_json.py ├── test_io_parquet.py └── test_io_path.py ├── jobs ├── test_jobs_mlflow_save_configs.py ├── test_jobs_optimize_saved_model.py └── test_jobs_save_dataset.py ├── layers ├── test_layers_base.py ├── test_layers_bpr.py ├── test_layers_bpr_max.py ├── test_layers_combinators.py ├── test_layers_core.py ├── test_layers_embeddings.py ├── test_layers_lookup.py ├── test_layers_lstm.py ├── test_layers_mask.py ├── test_layers_multi.py ├── test_layers_multi_css.py ├── test_layers_nce_loss.py ├── test_layers_reduce.py ├── test_layers_size.py ├── test_layers_slice.py ├── test_layers_sparse.py ├── test_layers_string.py ├── test_layers_top_one.py ├── test_layers_top_one_max.py └── test_layers_transformer.py ├── prepros ├── test_prepros_base.py ├── test_prepros_combinators.py ├── test_prepros_core.py └── test_prepros_record.py ├── readers ├── test_readers_generator.py └── test_readers_record.py ├── utils ├── test_utils_broadcasting.py ├── test_utils_datastruct.py ├── test_utils_field.py ├── test_utils_iter.py ├── test_utils_tables.py └── test_utils_uuid.py └── writers └── test_writers_record.py /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.github/workflows/doc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/.github/workflows/doc.yml -------------------------------------------------------------------------------- /.github/workflows/publish-to-test-pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/.github/workflows/publish-to-test-pypi.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/.pylintrc -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/Makefile -------------------------------------------------------------------------------- /check_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/check_version.py -------------------------------------------------------------------------------- /deepr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/__init__.py -------------------------------------------------------------------------------- /deepr/cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/cli/__init__.py -------------------------------------------------------------------------------- /deepr/cli/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/cli/main.py -------------------------------------------------------------------------------- /deepr/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/config/__init__.py -------------------------------------------------------------------------------- /deepr/config/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/config/base.py -------------------------------------------------------------------------------- /deepr/config/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/config/experimental.py -------------------------------------------------------------------------------- /deepr/config/macros.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/config/macros.py -------------------------------------------------------------------------------- /deepr/config/references.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/config/references.py -------------------------------------------------------------------------------- /deepr/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepr/examples/movielens/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/__init__.py -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/README.md -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/average/config.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/average/config.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/average/macros.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/average/macros.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/average_svd/config.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/average_svd/config.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/average_svd/macros.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/average_svd/macros.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/common/end.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/common/end.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/common/evaluate.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/common/evaluate.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/common/macros.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/common/macros.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/common/start.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/common/start.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/common/train.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/common/train.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/vae/config.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/vae/config.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/vae/macros.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/vae/macros.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/vae_svd/config.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/vae_svd/config.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/configs/vae_svd/macros.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/configs/vae_svd/macros.jsonnet -------------------------------------------------------------------------------- /deepr/examples/movielens/jobs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/jobs/__init__.py -------------------------------------------------------------------------------- /deepr/examples/movielens/jobs/build_records.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/jobs/build_records.py -------------------------------------------------------------------------------- /deepr/examples/movielens/jobs/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/jobs/evaluate.py -------------------------------------------------------------------------------- /deepr/examples/movielens/jobs/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/jobs/init.py -------------------------------------------------------------------------------- /deepr/examples/movielens/jobs/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/jobs/predict.py -------------------------------------------------------------------------------- /deepr/examples/movielens/jobs/svd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/jobs/svd.py -------------------------------------------------------------------------------- /deepr/examples/movielens/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/layers/__init__.py -------------------------------------------------------------------------------- /deepr/examples/movielens/layers/average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/layers/average.py -------------------------------------------------------------------------------- /deepr/examples/movielens/layers/bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/layers/bpr.py -------------------------------------------------------------------------------- /deepr/examples/movielens/layers/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/layers/losses.py -------------------------------------------------------------------------------- /deepr/examples/movielens/layers/multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/layers/multi.py -------------------------------------------------------------------------------- /deepr/examples/movielens/layers/ns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/layers/ns.py -------------------------------------------------------------------------------- /deepr/examples/movielens/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/layers/transformer.py -------------------------------------------------------------------------------- /deepr/examples/movielens/layers/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/layers/vae.py -------------------------------------------------------------------------------- /deepr/examples/movielens/macros/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/macros/__init__.py -------------------------------------------------------------------------------- /deepr/examples/movielens/macros/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/macros/paths.py -------------------------------------------------------------------------------- /deepr/examples/movielens/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/metrics/__init__.py -------------------------------------------------------------------------------- /deepr/examples/movielens/metrics/ndcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/metrics/ndcg.py -------------------------------------------------------------------------------- /deepr/examples/movielens/metrics/recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/metrics/recall.py -------------------------------------------------------------------------------- /deepr/examples/movielens/prepros/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/prepros/__init__.py -------------------------------------------------------------------------------- /deepr/examples/movielens/prepros/csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/prepros/csv.py -------------------------------------------------------------------------------- /deepr/examples/movielens/prepros/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/prepros/record.py -------------------------------------------------------------------------------- /deepr/examples/movielens/readers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/readers/__init__.py -------------------------------------------------------------------------------- /deepr/examples/movielens/readers/csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/readers/csv.py -------------------------------------------------------------------------------- /deepr/examples/movielens/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deepr/examples/movielens/utils/fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/movielens/utils/fields.py -------------------------------------------------------------------------------- /deepr/examples/multiply/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/__init__.py -------------------------------------------------------------------------------- /deepr/examples/multiply/configs/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/configs/config.json -------------------------------------------------------------------------------- /deepr/examples/multiply/configs/macros.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/configs/macros.json -------------------------------------------------------------------------------- /deepr/examples/multiply/jobs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/jobs/__init__.py -------------------------------------------------------------------------------- /deepr/examples/multiply/jobs/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/jobs/build.py -------------------------------------------------------------------------------- /deepr/examples/multiply/jobs/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/jobs/predict.py -------------------------------------------------------------------------------- /deepr/examples/multiply/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/layers/__init__.py -------------------------------------------------------------------------------- /deepr/examples/multiply/layers/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/layers/loss.py -------------------------------------------------------------------------------- /deepr/examples/multiply/layers/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/layers/model.py -------------------------------------------------------------------------------- /deepr/examples/multiply/macros/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/macros/__init__.py -------------------------------------------------------------------------------- /deepr/examples/multiply/macros/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/macros/paths.py -------------------------------------------------------------------------------- /deepr/examples/multiply/prepros/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/prepros/__init__.py -------------------------------------------------------------------------------- /deepr/examples/multiply/prepros/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/prepros/default.py -------------------------------------------------------------------------------- /deepr/examples/multiply/prepros/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/examples/multiply/prepros/inference.py -------------------------------------------------------------------------------- /deepr/exporters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/exporters/__init__.py -------------------------------------------------------------------------------- /deepr/exporters/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/exporters/base.py -------------------------------------------------------------------------------- /deepr/exporters/best_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/exporters/best_checkpoint.py -------------------------------------------------------------------------------- /deepr/exporters/save_variables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/exporters/save_variables.py -------------------------------------------------------------------------------- /deepr/exporters/saved_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/exporters/saved_model.py -------------------------------------------------------------------------------- /deepr/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/hooks/__init__.py -------------------------------------------------------------------------------- /deepr/hooks/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/hooks/base.py -------------------------------------------------------------------------------- /deepr/hooks/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/hooks/early_stopping.py -------------------------------------------------------------------------------- /deepr/hooks/log_variables_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/hooks/log_variables_init.py -------------------------------------------------------------------------------- /deepr/hooks/logging_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/hooks/logging_tensor.py -------------------------------------------------------------------------------- /deepr/hooks/num_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/hooks/num_params.py -------------------------------------------------------------------------------- /deepr/hooks/steps_per_sec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/hooks/steps_per_sec.py -------------------------------------------------------------------------------- /deepr/hooks/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/hooks/summary.py -------------------------------------------------------------------------------- /deepr/initializers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/initializers/__init__.py -------------------------------------------------------------------------------- /deepr/initializers/checkpoint_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/initializers/checkpoint_initializer.py -------------------------------------------------------------------------------- /deepr/io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/io/__init__.py -------------------------------------------------------------------------------- /deepr/io/hdfs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/io/hdfs.py -------------------------------------------------------------------------------- /deepr/io/json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/io/json.py -------------------------------------------------------------------------------- /deepr/io/parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/io/parquet.py -------------------------------------------------------------------------------- /deepr/io/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/io/path.py -------------------------------------------------------------------------------- /deepr/jobs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/__init__.py -------------------------------------------------------------------------------- /deepr/jobs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/base.py -------------------------------------------------------------------------------- /deepr/jobs/cleanup_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/cleanup_checkpoints.py -------------------------------------------------------------------------------- /deepr/jobs/combinators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/combinators.py -------------------------------------------------------------------------------- /deepr/jobs/copy_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/copy_dir.py -------------------------------------------------------------------------------- /deepr/jobs/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/evaluate.py -------------------------------------------------------------------------------- /deepr/jobs/export_xla_model_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/export_xla_model_metadata.py -------------------------------------------------------------------------------- /deepr/jobs/log_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/log_metric.py -------------------------------------------------------------------------------- /deepr/jobs/mlflow_save_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/mlflow_save_configs.py -------------------------------------------------------------------------------- /deepr/jobs/mlflow_save_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/mlflow_save_info.py -------------------------------------------------------------------------------- /deepr/jobs/optimize_saved_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/optimize_saved_model.py -------------------------------------------------------------------------------- /deepr/jobs/params_tuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/params_tuner.py -------------------------------------------------------------------------------- /deepr/jobs/save_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/save_dataset.py -------------------------------------------------------------------------------- /deepr/jobs/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/trainer.py -------------------------------------------------------------------------------- /deepr/jobs/trainer_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/trainer_base.py -------------------------------------------------------------------------------- /deepr/jobs/trainer_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/trainer_keras.py -------------------------------------------------------------------------------- /deepr/jobs/yarn_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/yarn_config.py -------------------------------------------------------------------------------- /deepr/jobs/yarn_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/yarn_launcher.py -------------------------------------------------------------------------------- /deepr/jobs/yarn_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/jobs/yarn_trainer.py -------------------------------------------------------------------------------- /deepr/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/__init__.py -------------------------------------------------------------------------------- /deepr/layers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/base.py -------------------------------------------------------------------------------- /deepr/layers/bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/bpr.py -------------------------------------------------------------------------------- /deepr/layers/bpr_max.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/bpr_max.py -------------------------------------------------------------------------------- /deepr/layers/click_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/click_rank.py -------------------------------------------------------------------------------- /deepr/layers/combinators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/combinators.py -------------------------------------------------------------------------------- /deepr/layers/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/core.py -------------------------------------------------------------------------------- /deepr/layers/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/dropout.py -------------------------------------------------------------------------------- /deepr/layers/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/embedding.py -------------------------------------------------------------------------------- /deepr/layers/lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/lookup.py -------------------------------------------------------------------------------- /deepr/layers/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/lstm.py -------------------------------------------------------------------------------- /deepr/layers/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/mask.py -------------------------------------------------------------------------------- /deepr/layers/multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/multi.py -------------------------------------------------------------------------------- /deepr/layers/multi_css.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/multi_css.py -------------------------------------------------------------------------------- /deepr/layers/nce_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/nce_loss.py -------------------------------------------------------------------------------- /deepr/layers/reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/reduce.py -------------------------------------------------------------------------------- /deepr/layers/size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/size.py -------------------------------------------------------------------------------- /deepr/layers/slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/slice.py -------------------------------------------------------------------------------- /deepr/layers/sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/sparse.py -------------------------------------------------------------------------------- /deepr/layers/string.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/string.py -------------------------------------------------------------------------------- /deepr/layers/top_one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/top_one.py -------------------------------------------------------------------------------- /deepr/layers/top_one_max.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/top_one_max.py -------------------------------------------------------------------------------- /deepr/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/transformer.py -------------------------------------------------------------------------------- /deepr/layers/triplet_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/layers/triplet_precision.py -------------------------------------------------------------------------------- /deepr/macros/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/macros/__init__.py -------------------------------------------------------------------------------- /deepr/macros/dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/macros/dynamic.py -------------------------------------------------------------------------------- /deepr/macros/graphite_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/macros/graphite_init.py -------------------------------------------------------------------------------- /deepr/macros/mlflow_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/macros/mlflow_init.py -------------------------------------------------------------------------------- /deepr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/metrics/__init__.py -------------------------------------------------------------------------------- /deepr/metrics/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/metrics/accuracy.py -------------------------------------------------------------------------------- /deepr/metrics/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/metrics/base.py -------------------------------------------------------------------------------- /deepr/metrics/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/metrics/core.py -------------------------------------------------------------------------------- /deepr/metrics/mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/metrics/mean.py -------------------------------------------------------------------------------- /deepr/metrics/step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/metrics/step.py -------------------------------------------------------------------------------- /deepr/metrics/variable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/metrics/variable.py -------------------------------------------------------------------------------- /deepr/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/optimizers/__init__.py -------------------------------------------------------------------------------- /deepr/optimizers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/optimizers/base.py -------------------------------------------------------------------------------- /deepr/optimizers/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/optimizers/core.py -------------------------------------------------------------------------------- /deepr/predictors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/predictors/__init__.py -------------------------------------------------------------------------------- /deepr/predictors/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/predictors/base.py -------------------------------------------------------------------------------- /deepr/predictors/proto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/predictors/proto.py -------------------------------------------------------------------------------- /deepr/predictors/saved_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/predictors/saved_model.py -------------------------------------------------------------------------------- /deepr/prepros/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/prepros/__init__.py -------------------------------------------------------------------------------- /deepr/prepros/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/prepros/base.py -------------------------------------------------------------------------------- /deepr/prepros/combinators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/prepros/combinators.py -------------------------------------------------------------------------------- /deepr/prepros/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/prepros/core.py -------------------------------------------------------------------------------- /deepr/prepros/lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/prepros/lookup.py -------------------------------------------------------------------------------- /deepr/prepros/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/prepros/record.py -------------------------------------------------------------------------------- /deepr/readers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/readers/__init__.py -------------------------------------------------------------------------------- /deepr/readers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/readers/base.py -------------------------------------------------------------------------------- /deepr/readers/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/readers/generator.py -------------------------------------------------------------------------------- /deepr/readers/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/readers/record.py -------------------------------------------------------------------------------- /deepr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/__init__.py -------------------------------------------------------------------------------- /deepr/utils/broadcasting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/broadcasting.py -------------------------------------------------------------------------------- /deepr/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/checkpoint.py -------------------------------------------------------------------------------- /deepr/utils/datastruct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/datastruct.py -------------------------------------------------------------------------------- /deepr/utils/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/exceptions.py -------------------------------------------------------------------------------- /deepr/utils/field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/field.py -------------------------------------------------------------------------------- /deepr/utils/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/graph.py -------------------------------------------------------------------------------- /deepr/utils/graphite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/graphite.py -------------------------------------------------------------------------------- /deepr/utils/iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/iter.py -------------------------------------------------------------------------------- /deepr/utils/mlflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/mlflow.py -------------------------------------------------------------------------------- /deepr/utils/tables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/tables.py -------------------------------------------------------------------------------- /deepr/utils/tf2xla_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/tf2xla_pb2.py -------------------------------------------------------------------------------- /deepr/utils/uuid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/utils/uuid.py -------------------------------------------------------------------------------- /deepr/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/version.py -------------------------------------------------------------------------------- /deepr/vocab/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/vocab/__init__.py -------------------------------------------------------------------------------- /deepr/vocab/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/vocab/base.py -------------------------------------------------------------------------------- /deepr/writers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/writers/__init__.py -------------------------------------------------------------------------------- /deepr/writers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/writers/base.py -------------------------------------------------------------------------------- /deepr/writers/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/deepr/writers/record.py -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | deepr* 3 | -------------------------------------------------------------------------------- /docs/API/core.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/API/core.rst -------------------------------------------------------------------------------- /docs/CHANGELOG.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/CHANGELOG.rst -------------------------------------------------------------------------------- /docs/CONTRIBUTING.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/CONTRIBUTING.rst -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/README.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/getting_started/advanced.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/getting_started/advanced.ipynb -------------------------------------------------------------------------------- /docs/getting_started/config.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/getting_started/config.ipynb -------------------------------------------------------------------------------- /docs/getting_started/pipeline.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/getting_started/pipeline.ipynb -------------------------------------------------------------------------------- /docs/getting_started/quickstart.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/getting_started/quickstart.ipynb -------------------------------------------------------------------------------- /docs/getting_started/tuning.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/getting_started/tuning.ipynb -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/movielens/movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/docs/movielens/movielens.ipynb -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/mypy.ini -------------------------------------------------------------------------------- /requirements-gpu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/requirements-gpu.txt -------------------------------------------------------------------------------- /requirements-test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/requirements-test.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/setup.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/integration/examples/multiply/test_example_multiply_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/integration/examples/multiply/test_example_multiply_configs.py -------------------------------------------------------------------------------- /tests/integration/examples/multiply/test_example_multiply_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/integration/examples/multiply/test_example_multiply_pipeline.py -------------------------------------------------------------------------------- /tests/unit/cli/test_cli_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/cli/test_cli_main.py -------------------------------------------------------------------------------- /tests/unit/config/test_config_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/config/test_config_base.py -------------------------------------------------------------------------------- /tests/unit/config/test_config_experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/config/test_config_experimental.py -------------------------------------------------------------------------------- /tests/unit/config/test_config_macros.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/config/test_config_macros.py -------------------------------------------------------------------------------- /tests/unit/config/test_config_references.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/config/test_config_references.py -------------------------------------------------------------------------------- /tests/unit/exporters/test_exporters_best_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/exporters/test_exporters_best_checkpoint.py -------------------------------------------------------------------------------- /tests/unit/exporters/test_exporters_save_variables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/exporters/test_exporters_save_variables.py -------------------------------------------------------------------------------- /tests/unit/io/test_io_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/io/test_io_json.py -------------------------------------------------------------------------------- /tests/unit/io/test_io_parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/io/test_io_parquet.py -------------------------------------------------------------------------------- /tests/unit/io/test_io_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/io/test_io_path.py -------------------------------------------------------------------------------- /tests/unit/jobs/test_jobs_mlflow_save_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/jobs/test_jobs_mlflow_save_configs.py -------------------------------------------------------------------------------- /tests/unit/jobs/test_jobs_optimize_saved_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/jobs/test_jobs_optimize_saved_model.py -------------------------------------------------------------------------------- /tests/unit/jobs/test_jobs_save_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/jobs/test_jobs_save_dataset.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_base.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_bpr.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_bpr_max.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_bpr_max.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_combinators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_combinators.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_core.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_embeddings.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_lookup.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_lstm.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_mask.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_multi.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_multi_css.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_multi_css.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_nce_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_nce_loss.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_reduce.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_size.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_slice.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_sparse.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_string.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_string.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_top_one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_top_one.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_top_one_max.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_top_one_max.py -------------------------------------------------------------------------------- /tests/unit/layers/test_layers_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/layers/test_layers_transformer.py -------------------------------------------------------------------------------- /tests/unit/prepros/test_prepros_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/prepros/test_prepros_base.py -------------------------------------------------------------------------------- /tests/unit/prepros/test_prepros_combinators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/prepros/test_prepros_combinators.py -------------------------------------------------------------------------------- /tests/unit/prepros/test_prepros_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/prepros/test_prepros_core.py -------------------------------------------------------------------------------- /tests/unit/prepros/test_prepros_record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/prepros/test_prepros_record.py -------------------------------------------------------------------------------- /tests/unit/readers/test_readers_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/readers/test_readers_generator.py -------------------------------------------------------------------------------- /tests/unit/readers/test_readers_record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/readers/test_readers_record.py -------------------------------------------------------------------------------- /tests/unit/utils/test_utils_broadcasting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/utils/test_utils_broadcasting.py -------------------------------------------------------------------------------- /tests/unit/utils/test_utils_datastruct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/utils/test_utils_datastruct.py -------------------------------------------------------------------------------- /tests/unit/utils/test_utils_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/utils/test_utils_field.py -------------------------------------------------------------------------------- /tests/unit/utils/test_utils_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/utils/test_utils_iter.py -------------------------------------------------------------------------------- /tests/unit/utils/test_utils_tables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/utils/test_utils_tables.py -------------------------------------------------------------------------------- /tests/unit/utils/test_utils_uuid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/utils/test_utils_uuid.py -------------------------------------------------------------------------------- /tests/unit/writers/test_writers_record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/criteo/deepr/HEAD/tests/unit/writers/test_writers_record.py --------------------------------------------------------------------------------