├── .coveragerc ├── .dockerignore ├── .github └── PULL_REQUEST_TEMPLATE.md ├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── NOTICE ├── README.rst ├── docker ├── 1.7-1-1 │ ├── base │ │ └── Dockerfile.cpu │ ├── final │ │ └── Dockerfile.cpu │ └── resources │ │ └── mms │ │ ├── ExecutionParameters.java │ │ ├── config.properties.tmp │ │ └── endpoints-1.0.jar ├── 3.0-5 │ ├── base │ │ └── Dockerfile.cpu │ ├── final │ │ └── Dockerfile.cpu │ └── resources │ │ └── mms │ │ ├── ExecutionParameters.java │ │ ├── config.properties.tmp │ │ └── endpoints-1.0.jar └── configs │ └── dask_configs.yaml ├── pyproject.toml ├── requirements.txt ├── setup.cfg ├── setup.py ├── src ├── sagemaker_algorithm_toolkit │ ├── __init__.py │ ├── channel_validation.py │ ├── exceptions.py │ ├── hyperparameter_validation.py │ ├── metadata.py │ └── metrics.py └── sagemaker_xgboost_container │ ├── .DS_Store │ ├── __init__.py │ ├── algorithm_mode │ ├── __init__.py │ ├── channel_validation.py │ ├── handler_service.py │ ├── hyperparameter_validation.py │ ├── inference_errors.py │ ├── integration.py │ ├── metadata.py │ ├── metrics.py │ ├── serve.py │ ├── serve_utils.py │ ├── train.py │ └── train_utils.py │ ├── callback.py │ ├── checkpointing.py │ ├── constants │ ├── __init__.py │ ├── sm_env_constants.py │ ├── xgb_constants.py │ └── xgb_content_types.py │ ├── data_utils.py │ ├── distributed.py │ ├── distributed_gpu │ ├── __init__.py │ ├── dask_cluster_utils.py │ ├── dask_data_utils.py │ └── distributed_gpu_training.py │ ├── dmlc_patch │ ├── __init__.py │ └── tracker.py │ ├── encoder.py │ ├── handler_service.py │ ├── metrics │ ├── __init__.py │ └── custom_metrics.py │ ├── mms_patch │ ├── __init__.py │ ├── mms_transformer.py │ └── model_server.py │ ├── prediction_utils.py │ ├── serving.py │ ├── serving_mms.py │ └── training.py ├── test-requirements.txt ├── test ├── __init__.py ├── conftest.py ├── integration │ ├── local │ │ ├── __init__.py │ │ ├── test_abalone.py │ │ ├── test_boston.py │ │ ├── test_early_stopping.py │ │ ├── test_kfold.py │ │ ├── test_multiple_model_endpoint.py │ │ └── test_versions.py │ └── test_metadata_calls.py ├── resources │ ├── abalone-binary │ │ └── data │ │ │ ├── train │ │ │ ├── abalone.train_0 │ │ │ └── abalone.train_1 │ │ │ └── validation │ │ │ └── abalone.validation │ ├── abalone-multiclass │ │ └── data │ │ │ ├── train │ │ │ ├── abalone.train_0 │ │ │ └── abalone.train_1 │ │ │ └── validation │ │ │ └── abalone.validation │ ├── abalone-subdirs │ │ ├── dir1 │ │ │ └── dir2 │ │ │ │ └── dir3 │ │ │ │ └── dir4 │ │ │ │ └── abalone.train_0 │ │ ├── train │ │ │ ├── input_1 │ │ │ │ └── abalone.train_0 │ │ │ └── input_2 │ │ │ │ └── abalone.train_1 │ │ └── validation │ │ │ └── abalone.validation │ ├── abalone │ │ ├── abalone_distributed.py │ │ ├── data │ │ │ ├── train │ │ │ │ ├── abalone.train_0 │ │ │ │ └── abalone.train_1 │ │ │ └── validation │ │ │ │ └── abalone.validation │ │ └── models │ │ │ └── libsvm_pickled │ │ │ └── xgboost-model │ ├── boston │ │ ├── data │ │ │ └── empty │ │ └── single_machine_customer_script.py │ ├── data │ │ ├── csv │ │ │ ├── csv_files │ │ │ │ └── train.csv │ │ │ ├── multiple_files │ │ │ │ ├── train_1.csv │ │ │ │ └── train_2.csv │ │ │ ├── train.csv │ │ │ ├── train.csv.weights │ │ │ └── weighted_csv_files │ │ │ │ └── train.csv.weights │ │ ├── libsvm │ │ │ ├── libsvm_files │ │ │ │ └── train.libsvm │ │ │ ├── train.libsvm │ │ │ └── train.libsvm.weights │ │ ├── parquet │ │ │ ├── multiple_files │ │ │ │ ├── train_0.parquet │ │ │ │ └── train_1.parquet │ │ │ ├── pq_files │ │ │ │ └── train.parquet │ │ │ └── train.parquet │ │ └── recordio_protobuf │ │ │ ├── pb_files │ │ │ └── train.pb │ │ │ ├── single_feature_label.pb │ │ │ ├── sparse │ │ │ └── train.pb │ │ │ ├── sparse_edge_cases │ │ │ ├── dense_as_sparse.pbr │ │ │ ├── diagonal.pbr │ │ │ ├── rectangular_sparse.pbr │ │ │ ├── single_value_bot_left.pbr │ │ │ ├── single_value_bot_right.pbr │ │ │ ├── single_value_center.pbr │ │ │ ├── single_value_top_left.pbr │ │ │ └── single_value_top_right.pbr │ │ │ └── train.pb │ ├── early_stopping │ │ └── data │ │ │ └── train │ │ │ ├── abalone.train_0 │ │ │ └── abalone.train_1 │ ├── models │ │ ├── pickled_model │ │ │ └── xgboost-model │ │ └── saved_booster │ │ │ └── xgboost-model │ └── versions │ │ └── train.py ├── unit │ ├── .DS_Store │ ├── algorithm_mode │ │ ├── __init__.py │ │ ├── test_algorithm_mode.py │ │ ├── test_channel_validation.py │ │ ├── test_custom_metrics.py │ │ ├── test_hyperparameter_validation.py │ │ ├── test_serve.py │ │ ├── test_serve_utils.py │ │ └── test_train_utils.py │ ├── algorithm_toolkit │ │ ├── __init__.py │ │ ├── test_channel_validation.py │ │ ├── test_exceptions.py │ │ ├── test_hyperparameter_validation.py │ │ └── test_metrics.py │ ├── distributed_gpu │ │ ├── __init__.py │ │ ├── test_dask_data_utils.py │ │ └── test_distributed_gpu_training.py │ ├── test_checkpointing.py │ ├── test_data_utils.py │ ├── test_distributed.py │ ├── test_encoder.py │ ├── test_handler_service.py │ ├── test_prediction_utils.py │ ├── test_serving.py │ ├── test_serving_mms.py │ └── test_training.py └── utils │ ├── __init__.py │ ├── local_mode.py │ ├── sagemaker_pipe.py │ └── test_utils.py └── tox.ini /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/.coveragerc -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- 1 | .venv 2 | .tox 3 | -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/NOTICE -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/README.rst -------------------------------------------------------------------------------- /docker/1.7-1-1/base/Dockerfile.cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/1.7-1-1/base/Dockerfile.cpu -------------------------------------------------------------------------------- /docker/1.7-1-1/final/Dockerfile.cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/1.7-1-1/final/Dockerfile.cpu -------------------------------------------------------------------------------- /docker/1.7-1-1/resources/mms/ExecutionParameters.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/1.7-1-1/resources/mms/ExecutionParameters.java -------------------------------------------------------------------------------- /docker/1.7-1-1/resources/mms/config.properties.tmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/1.7-1-1/resources/mms/config.properties.tmp -------------------------------------------------------------------------------- /docker/1.7-1-1/resources/mms/endpoints-1.0.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/1.7-1-1/resources/mms/endpoints-1.0.jar -------------------------------------------------------------------------------- /docker/3.0-5/base/Dockerfile.cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/3.0-5/base/Dockerfile.cpu -------------------------------------------------------------------------------- /docker/3.0-5/final/Dockerfile.cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/3.0-5/final/Dockerfile.cpu -------------------------------------------------------------------------------- /docker/3.0-5/resources/mms/ExecutionParameters.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/3.0-5/resources/mms/ExecutionParameters.java -------------------------------------------------------------------------------- /docker/3.0-5/resources/mms/config.properties.tmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/3.0-5/resources/mms/config.properties.tmp -------------------------------------------------------------------------------- /docker/3.0-5/resources/mms/endpoints-1.0.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/3.0-5/resources/mms/endpoints-1.0.jar -------------------------------------------------------------------------------- /docker/configs/dask_configs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/docker/configs/dask_configs.yaml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/setup.py -------------------------------------------------------------------------------- /src/sagemaker_algorithm_toolkit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_algorithm_toolkit/__init__.py -------------------------------------------------------------------------------- /src/sagemaker_algorithm_toolkit/channel_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_algorithm_toolkit/channel_validation.py -------------------------------------------------------------------------------- /src/sagemaker_algorithm_toolkit/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_algorithm_toolkit/exceptions.py -------------------------------------------------------------------------------- /src/sagemaker_algorithm_toolkit/hyperparameter_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_algorithm_toolkit/hyperparameter_validation.py -------------------------------------------------------------------------------- /src/sagemaker_algorithm_toolkit/metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_algorithm_toolkit/metadata.py -------------------------------------------------------------------------------- /src/sagemaker_algorithm_toolkit/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_algorithm_toolkit/metrics.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/.DS_Store -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/__init__.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/__init__.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/channel_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/channel_validation.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/handler_service.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/handler_service.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/hyperparameter_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/hyperparameter_validation.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/inference_errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/inference_errors.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/integration.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/metadata.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/metrics.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/serve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/serve.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/serve_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/serve_utils.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/train.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/algorithm_mode/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/algorithm_mode/train_utils.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/callback.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/checkpointing.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/constants/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/constants/__init__.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/constants/sm_env_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/constants/sm_env_constants.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/constants/xgb_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/constants/xgb_constants.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/constants/xgb_content_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/constants/xgb_content_types.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/data_utils.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/distributed.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/distributed_gpu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/distributed_gpu/dask_cluster_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/distributed_gpu/dask_cluster_utils.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/distributed_gpu/dask_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/distributed_gpu/dask_data_utils.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/distributed_gpu/distributed_gpu_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/distributed_gpu/distributed_gpu_training.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/dmlc_patch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/dmlc_patch/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/dmlc_patch/tracker.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/encoder.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/handler_service.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/handler_service.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/metrics/__init__.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/metrics/custom_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/metrics/custom_metrics.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/mms_patch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/mms_patch/__init__.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/mms_patch/mms_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/mms_patch/mms_transformer.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/mms_patch/model_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/mms_patch/model_server.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/prediction_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/prediction_utils.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/serving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/serving.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/serving_mms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/serving_mms.py -------------------------------------------------------------------------------- /src/sagemaker_xgboost_container/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/src/sagemaker_xgboost_container/training.py -------------------------------------------------------------------------------- /test-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test-requirements.txt -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/__init__.py -------------------------------------------------------------------------------- /test/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/conftest.py -------------------------------------------------------------------------------- /test/integration/local/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/integration/local/__init__.py -------------------------------------------------------------------------------- /test/integration/local/test_abalone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/integration/local/test_abalone.py -------------------------------------------------------------------------------- /test/integration/local/test_boston.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/integration/local/test_boston.py -------------------------------------------------------------------------------- /test/integration/local/test_early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/integration/local/test_early_stopping.py -------------------------------------------------------------------------------- /test/integration/local/test_kfold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/integration/local/test_kfold.py -------------------------------------------------------------------------------- /test/integration/local/test_multiple_model_endpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/integration/local/test_multiple_model_endpoint.py -------------------------------------------------------------------------------- /test/integration/local/test_versions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/integration/local/test_versions.py -------------------------------------------------------------------------------- /test/integration/test_metadata_calls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/integration/test_metadata_calls.py -------------------------------------------------------------------------------- /test/resources/abalone-binary/data/train/abalone.train_0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-binary/data/train/abalone.train_0 -------------------------------------------------------------------------------- /test/resources/abalone-binary/data/train/abalone.train_1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-binary/data/train/abalone.train_1 -------------------------------------------------------------------------------- /test/resources/abalone-binary/data/validation/abalone.validation: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-binary/data/validation/abalone.validation -------------------------------------------------------------------------------- /test/resources/abalone-multiclass/data/train/abalone.train_0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-multiclass/data/train/abalone.train_0 -------------------------------------------------------------------------------- /test/resources/abalone-multiclass/data/train/abalone.train_1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-multiclass/data/train/abalone.train_1 -------------------------------------------------------------------------------- /test/resources/abalone-multiclass/data/validation/abalone.validation: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-multiclass/data/validation/abalone.validation -------------------------------------------------------------------------------- /test/resources/abalone-subdirs/dir1/dir2/dir3/dir4/abalone.train_0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-subdirs/dir1/dir2/dir3/dir4/abalone.train_0 -------------------------------------------------------------------------------- /test/resources/abalone-subdirs/train/input_1/abalone.train_0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-subdirs/train/input_1/abalone.train_0 -------------------------------------------------------------------------------- /test/resources/abalone-subdirs/train/input_2/abalone.train_1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-subdirs/train/input_2/abalone.train_1 -------------------------------------------------------------------------------- /test/resources/abalone-subdirs/validation/abalone.validation: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone-subdirs/validation/abalone.validation -------------------------------------------------------------------------------- /test/resources/abalone/abalone_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone/abalone_distributed.py -------------------------------------------------------------------------------- /test/resources/abalone/data/train/abalone.train_0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone/data/train/abalone.train_0 -------------------------------------------------------------------------------- /test/resources/abalone/data/train/abalone.train_1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone/data/train/abalone.train_1 -------------------------------------------------------------------------------- /test/resources/abalone/data/validation/abalone.validation: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone/data/validation/abalone.validation -------------------------------------------------------------------------------- /test/resources/abalone/models/libsvm_pickled/xgboost-model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/abalone/models/libsvm_pickled/xgboost-model -------------------------------------------------------------------------------- /test/resources/boston/data/empty: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/resources/boston/single_machine_customer_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/boston/single_machine_customer_script.py -------------------------------------------------------------------------------- /test/resources/data/csv/csv_files/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/csv/csv_files/train.csv -------------------------------------------------------------------------------- /test/resources/data/csv/multiple_files/train_1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/csv/multiple_files/train_1.csv -------------------------------------------------------------------------------- /test/resources/data/csv/multiple_files/train_2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/csv/multiple_files/train_2.csv -------------------------------------------------------------------------------- /test/resources/data/csv/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/csv/train.csv -------------------------------------------------------------------------------- /test/resources/data/csv/train.csv.weights: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/csv/train.csv.weights -------------------------------------------------------------------------------- /test/resources/data/csv/weighted_csv_files/train.csv.weights: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/csv/weighted_csv_files/train.csv.weights -------------------------------------------------------------------------------- /test/resources/data/libsvm/libsvm_files/train.libsvm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/libsvm/libsvm_files/train.libsvm -------------------------------------------------------------------------------- /test/resources/data/libsvm/train.libsvm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/libsvm/train.libsvm -------------------------------------------------------------------------------- /test/resources/data/libsvm/train.libsvm.weights: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/libsvm/train.libsvm.weights -------------------------------------------------------------------------------- /test/resources/data/parquet/multiple_files/train_0.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/parquet/multiple_files/train_0.parquet -------------------------------------------------------------------------------- /test/resources/data/parquet/multiple_files/train_1.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/parquet/multiple_files/train_1.parquet -------------------------------------------------------------------------------- /test/resources/data/parquet/pq_files/train.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/parquet/pq_files/train.parquet -------------------------------------------------------------------------------- /test/resources/data/parquet/train.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/parquet/train.parquet -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/pb_files/train.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/pb_files/train.pb -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/single_feature_label.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/single_feature_label.pb -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse/train.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse/train.pb -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse_edge_cases/dense_as_sparse.pbr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse_edge_cases/dense_as_sparse.pbr -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse_edge_cases/diagonal.pbr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse_edge_cases/diagonal.pbr -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse_edge_cases/rectangular_sparse.pbr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse_edge_cases/rectangular_sparse.pbr -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_bot_left.pbr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_bot_left.pbr -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_bot_right.pbr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_bot_right.pbr -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_center.pbr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_center.pbr -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_top_left.pbr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_top_left.pbr -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_top_right.pbr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/sparse_edge_cases/single_value_top_right.pbr -------------------------------------------------------------------------------- /test/resources/data/recordio_protobuf/train.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/data/recordio_protobuf/train.pb -------------------------------------------------------------------------------- /test/resources/early_stopping/data/train/abalone.train_0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/early_stopping/data/train/abalone.train_0 -------------------------------------------------------------------------------- /test/resources/early_stopping/data/train/abalone.train_1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/early_stopping/data/train/abalone.train_1 -------------------------------------------------------------------------------- /test/resources/models/pickled_model/xgboost-model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/models/pickled_model/xgboost-model -------------------------------------------------------------------------------- /test/resources/models/saved_booster/xgboost-model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/models/saved_booster/xgboost-model -------------------------------------------------------------------------------- /test/resources/versions/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/resources/versions/train.py -------------------------------------------------------------------------------- /test/unit/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/.DS_Store -------------------------------------------------------------------------------- /test/unit/algorithm_mode/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_mode/__init__.py -------------------------------------------------------------------------------- /test/unit/algorithm_mode/test_algorithm_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_mode/test_algorithm_mode.py -------------------------------------------------------------------------------- /test/unit/algorithm_mode/test_channel_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_mode/test_channel_validation.py -------------------------------------------------------------------------------- /test/unit/algorithm_mode/test_custom_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_mode/test_custom_metrics.py -------------------------------------------------------------------------------- /test/unit/algorithm_mode/test_hyperparameter_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_mode/test_hyperparameter_validation.py -------------------------------------------------------------------------------- /test/unit/algorithm_mode/test_serve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_mode/test_serve.py -------------------------------------------------------------------------------- /test/unit/algorithm_mode/test_serve_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_mode/test_serve_utils.py -------------------------------------------------------------------------------- /test/unit/algorithm_mode/test_train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_mode/test_train_utils.py -------------------------------------------------------------------------------- /test/unit/algorithm_toolkit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/unit/algorithm_toolkit/test_channel_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_toolkit/test_channel_validation.py -------------------------------------------------------------------------------- /test/unit/algorithm_toolkit/test_exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_toolkit/test_exceptions.py -------------------------------------------------------------------------------- /test/unit/algorithm_toolkit/test_hyperparameter_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_toolkit/test_hyperparameter_validation.py -------------------------------------------------------------------------------- /test/unit/algorithm_toolkit/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/algorithm_toolkit/test_metrics.py -------------------------------------------------------------------------------- /test/unit/distributed_gpu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/unit/distributed_gpu/test_dask_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/distributed_gpu/test_dask_data_utils.py -------------------------------------------------------------------------------- /test/unit/distributed_gpu/test_distributed_gpu_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/distributed_gpu/test_distributed_gpu_training.py -------------------------------------------------------------------------------- /test/unit/test_checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_checkpointing.py -------------------------------------------------------------------------------- /test/unit/test_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_data_utils.py -------------------------------------------------------------------------------- /test/unit/test_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_distributed.py -------------------------------------------------------------------------------- /test/unit/test_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_encoder.py -------------------------------------------------------------------------------- /test/unit/test_handler_service.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_handler_service.py -------------------------------------------------------------------------------- /test/unit/test_prediction_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_prediction_utils.py -------------------------------------------------------------------------------- /test/unit/test_serving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_serving.py -------------------------------------------------------------------------------- /test/unit/test_serving_mms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_serving_mms.py -------------------------------------------------------------------------------- /test/unit/test_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/unit/test_training.py -------------------------------------------------------------------------------- /test/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/utils/__init__.py -------------------------------------------------------------------------------- /test/utils/local_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/utils/local_mode.py -------------------------------------------------------------------------------- /test/utils/sagemaker_pipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/utils/sagemaker_pipe.py -------------------------------------------------------------------------------- /test/utils/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/test/utils/test_utils.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws/sagemaker-xgboost-container/HEAD/tox.ini --------------------------------------------------------------------------------