gitextract_kt5qph6q/ ├── .clang-format ├── .clang-tidy ├── .editorconfig ├── .gitattributes ├── .github/ │ ├── FUNDING.yml │ ├── ISSUE_TEMPLATE.md │ ├── dependabot.yml │ ├── lock.yml │ ├── runs-on.yml │ └── workflows/ │ ├── cccl_nightly.yml │ ├── freebsd.yml │ ├── jvm_tests.yml │ ├── lint.yml │ ├── main.yml │ ├── misc.yml │ ├── pre-commit.yml │ ├── python_tests.yml │ ├── python_wheels_variants.yml │ ├── python_wheels_winarm64.yml │ ├── r_nold.yml │ ├── r_tests.yml │ ├── scorecards.yml │ ├── sycl_tests.yml │ └── windows.yml ├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── CITATION ├── CMakeLists.txt ├── CONTRIBUTORS.md ├── LICENSE ├── NEWS.md ├── R-package/ │ ├── .Rbuildignore │ ├── .gitignore │ ├── CMakeLists.txt │ ├── DESCRIPTION │ ├── NAMESPACE │ ├── R/ │ │ ├── callbacks.R │ │ ├── utils.R │ │ ├── xgb.Booster.R │ │ ├── xgb.DMatrix.R │ │ ├── xgb.DMatrix.save.R │ │ ├── xgb.config.R │ │ ├── xgb.create.features.R │ │ ├── xgb.cv.R │ │ ├── xgb.dump.R │ │ ├── xgb.ggplot.R │ │ ├── xgb.importance.R │ │ ├── xgb.load.R │ │ ├── xgb.load.raw.R │ │ ├── xgb.model.dt.tree.R │ │ ├── xgb.plot.deepness.R │ │ ├── xgb.plot.importance.R │ │ ├── xgb.plot.multi.trees.R │ │ ├── xgb.plot.shap.R │ │ ├── xgb.plot.tree.R │ │ ├── xgb.save.R │ │ ├── xgb.save.raw.R │ │ ├── xgb.train.R │ │ └── xgboost.R │ ├── README.md │ ├── bootstrap.R │ ├── cleanup │ ├── configure │ ├── configure.ac │ ├── configure.win │ ├── data/ │ │ ├── agaricus.test.rda │ │ └── agaricus.train.rda │ ├── inst/ │ │ └── make-r-def.R │ ├── man/ │ │ ├── a-compatibility-note-for-saveRDS-save.Rd │ │ ├── agaricus.test.Rd │ │ ├── agaricus.train.Rd │ │ ├── coef.xgb.Booster.Rd │ │ ├── dim.xgb.DMatrix.Rd │ │ ├── dimnames.xgb.DMatrix.Rd │ │ ├── getinfo.Rd │ │ ├── predict.xgb.Booster.Rd │ │ ├── predict.xgboost.Rd │ │ ├── print.xgb.Booster.Rd │ │ ├── print.xgb.DMatrix.Rd │ │ ├── print.xgb.cv.Rd │ │ ├── print.xgboost.Rd │ │ ├── variable.names.xgb.Booster.Rd │ │ ├── xgb.Callback.Rd │ │ ├── xgb.DMatrix.Rd │ │ ├── xgb.DMatrix.hasinfo.Rd │ │ ├── xgb.DMatrix.save.Rd │ │ ├── xgb.DataBatch.Rd │ │ ├── xgb.DataIter.Rd │ │ ├── xgb.ExtMemDMatrix.Rd │ │ ├── xgb.QuantileDMatrix.from_iterator.Rd │ │ ├── xgb.attr.Rd │ │ ├── xgb.cb.cv.predict.Rd │ │ ├── xgb.cb.early.stop.Rd │ │ ├── xgb.cb.evaluation.log.Rd │ │ ├── xgb.cb.gblinear.history.Rd │ │ ├── xgb.cb.print.evaluation.Rd │ │ ├── xgb.cb.reset.parameters.Rd │ │ ├── xgb.cb.save.model.Rd │ │ ├── xgb.config.Rd │ │ ├── xgb.copy.Booster.Rd │ │ ├── xgb.create.features.Rd │ │ ├── xgb.cv.Rd │ │ ├── xgb.dump.Rd │ │ ├── xgb.gblinear.history.Rd │ │ ├── xgb.get.DMatrix.data.Rd │ │ ├── xgb.get.DMatrix.num.non.missing.Rd │ │ ├── xgb.get.DMatrix.qcut.Rd │ │ ├── xgb.get.num.boosted.rounds.Rd │ │ ├── xgb.importance.Rd │ │ ├── xgb.is.same.Booster.Rd │ │ ├── xgb.load.Rd │ │ ├── xgb.load.raw.Rd │ │ ├── xgb.model.dt.tree.Rd │ │ ├── xgb.model.parameters.Rd │ │ ├── xgb.params.Rd │ │ ├── xgb.plot.deepness.Rd │ │ ├── xgb.plot.importance.Rd │ │ ├── xgb.plot.multi.trees.Rd │ │ ├── xgb.plot.shap.Rd │ │ ├── xgb.plot.shap.summary.Rd │ │ ├── xgb.plot.tree.Rd │ │ ├── xgb.save.Rd │ │ ├── xgb.save.raw.Rd │ │ ├── xgb.slice.Booster.Rd │ │ ├── xgb.slice.DMatrix.Rd │ │ ├── xgb.train.Rd │ │ ├── xgbConfig.Rd │ │ ├── xgboost-options.Rd │ │ └── xgboost.Rd │ ├── pkgdown/ │ │ └── _pkgdown.yml │ ├── remove_warning_suppression_pragma.sh │ ├── src/ │ │ ├── Makevars.in │ │ ├── Makevars.win.in │ │ ├── config.h.in │ │ ├── init.c │ │ ├── xgboost-win.def │ │ ├── xgboost_R.cc │ │ ├── xgboost_R.h │ │ └── xgboost_custom.cc │ ├── tests/ │ │ ├── helper_scripts/ │ │ │ ├── generate_models.R │ │ │ ├── install_deps.R │ │ │ └── run-examples.R │ │ ├── testthat/ │ │ │ ├── helper_model.R │ │ │ ├── test_basic.R │ │ │ ├── test_booster_slicing.R │ │ │ ├── test_callbacks.R │ │ │ ├── test_config.R │ │ │ ├── test_custom_objective.R │ │ │ ├── test_dmatrix.R │ │ │ ├── test_feature_weights.R │ │ │ ├── test_glm.R │ │ │ ├── test_helpers.R │ │ │ ├── test_interaction_constraints.R │ │ │ ├── test_interactions.R │ │ │ ├── test_io.R │ │ │ ├── test_model_compatibility.R │ │ │ ├── test_monotone.R │ │ │ ├── test_parameter_exposure.R │ │ │ ├── test_poisson_regression.R │ │ │ ├── test_ranking.R │ │ │ ├── test_unicode.R │ │ │ ├── test_update.R │ │ │ └── test_xgboost.R │ │ └── testthat.R │ └── vignettes/ │ ├── xgboost_introduction.Rmd │ └── xgboostfromJSON.Rmd ├── README.md ├── SECURITY.md ├── amalgamation/ │ └── dmlc-minimum0.cc ├── cmake/ │ ├── Doc.cmake │ ├── FindOpenMPMacOS.cmake │ ├── PrefetchIntrinsics.cmake │ ├── RPackageInstall.cmake.in │ ├── RPackageInstallTargetSetup.cmake │ ├── Sanitizer.cmake │ ├── Utils.cmake │ ├── Version.cmake │ ├── modules/ │ │ ├── FindASan.cmake │ │ ├── FindLSan.cmake │ │ ├── FindLibR.cmake │ │ ├── FindNVML.cmake │ │ ├── FindNccl.cmake │ │ ├── FindTSan.cmake │ │ └── FindUBSan.cmake │ ├── version_config.h.in │ ├── xgboost-config.cmake.in │ └── xgboost.pc.in ├── demo/ │ ├── .gitignore │ ├── README.md │ ├── aft_survival/ │ │ ├── README.rst │ │ ├── aft_survival_demo.py │ │ ├── aft_survival_demo_with_optuna.py │ │ └── aft_survival_viz_demo.py │ ├── c-api/ │ │ ├── .gitignore │ │ ├── basic/ │ │ │ ├── Makefile │ │ │ ├── README.md │ │ │ └── c-api-demo.c │ │ ├── external-memory/ │ │ │ ├── README.md │ │ │ └── external_memory.c │ │ └── inference/ │ │ └── inference.c │ ├── dask/ │ │ ├── README.rst │ │ ├── cpu_survival.py │ │ ├── cpu_training.py │ │ ├── dask_callbacks.py │ │ ├── dask_learning_to_rank.py │ │ ├── forward_logging.py │ │ ├── gpu_training.py │ │ ├── sklearn_cpu_training.py │ │ └── sklearn_gpu_training.py │ ├── data/ │ │ ├── README.md │ │ ├── gen_autoclaims.R │ │ └── regression/ │ │ ├── README.md │ │ ├── machine.names │ │ ├── mapfeat.py │ │ └── mknfold.py │ ├── guide-python/ │ │ ├── README.rst │ │ ├── basic_walkthrough.py │ │ ├── boost_from_prediction.py │ │ ├── callbacks.py │ │ ├── cat_in_the_dat.py │ │ ├── cat_pipeline.py │ │ ├── categorical.py │ │ ├── continuation.py │ │ ├── cover_type.py │ │ ├── cross_validation.py │ │ ├── custom_rmsle.py │ │ ├── custom_softmax.py │ │ ├── distributed_extmem_basic.py │ │ ├── evals_result.py │ │ ├── external_memory.py │ │ ├── feature_weights.py │ │ ├── gamma_regression.py │ │ ├── generalized_linear_model.py │ │ ├── gpu_tree_shap.py │ │ ├── individual_trees.py │ │ ├── learning_to_rank.py │ │ ├── model_parser.py │ │ ├── multioutput_reduced_gradient.py │ │ ├── multioutput_regression.py │ │ ├── predict_first_ntree.py │ │ ├── predict_leaf_indices.py │ │ ├── quantile_data_iterator.py │ │ ├── quantile_regression.py │ │ ├── sklearn_evals_result.py │ │ ├── sklearn_examples.py │ │ ├── sklearn_parallel.py │ │ ├── spark_estimator_examples.py │ │ └── update_process.py │ ├── kaggle-higgs/ │ │ ├── README.md │ │ ├── higgs-cv.py │ │ ├── higgs-numpy.py │ │ ├── higgs-pred.R │ │ ├── higgs-pred.py │ │ ├── higgs-train.R │ │ ├── run.sh │ │ ├── speedtest.R │ │ └── speedtest.py │ ├── kaggle-otto/ │ │ ├── README.MD │ │ ├── otto_train_pred.R │ │ └── understandingXGBoostModel.Rmd │ ├── multiclass_classification/ │ │ ├── README.md │ │ ├── runexp.sh │ │ ├── train.R │ │ └── train.py │ ├── nvflare/ │ │ ├── .gitignore │ │ ├── README.md │ │ ├── config/ │ │ │ ├── config_fed_client.json │ │ │ └── config_fed_server.json │ │ ├── horizontal/ │ │ │ ├── README.md │ │ │ ├── custom/ │ │ │ │ ├── controller.py │ │ │ │ └── trainer.py │ │ │ └── prepare_data.sh │ │ └── vertical/ │ │ ├── README.md │ │ ├── custom/ │ │ │ ├── controller.py │ │ │ └── trainer.py │ │ └── prepare_data.sh │ └── rmm_plugin/ │ ├── README.rst │ ├── rmm_mgpu_with_dask.py │ └── rmm_singlegpu.py ├── dev/ │ ├── prepare_jvm_release.py │ └── query_contributors.py ├── doc/ │ ├── .gitignore │ ├── Doxyfile.in │ ├── Makefile │ ├── R-package/ │ │ ├── .gitignore │ │ ├── Makefile │ │ ├── adding_parameters.rst │ │ ├── index.rst │ │ ├── index_base.rst │ │ ├── migration_guide.rst │ │ └── r_docs/ │ │ └── index.rst │ ├── README │ ├── build.rst │ ├── c++.rst │ ├── c.rst │ ├── changes/ │ │ ├── index.rst │ │ ├── v2.1.0.rst │ │ ├── v3.0.0.rst │ │ ├── v3.1.0.rst │ │ └── v3.2.0.rst │ ├── conf.py │ ├── contrib/ │ │ ├── ci.rst │ │ ├── coding_guide.rst │ │ ├── community.rst │ │ ├── consistency.rst │ │ ├── docs.rst │ │ ├── donate.rst │ │ ├── featuremap.rst │ │ ├── git_guide.rst │ │ ├── index.rst │ │ ├── python_packaging.rst │ │ ├── release.rst │ │ └── unit_tests.rst │ ├── dump.schema │ ├── faq.rst │ ├── get_started.rst │ ├── gpu/ │ │ └── index.rst │ ├── index.rst │ ├── install.rst │ ├── julia.rst │ ├── jvm/ │ │ ├── api.rst │ │ ├── index.rst │ │ ├── java_intro.rst │ │ ├── javadocs/ │ │ │ └── index.rst │ │ ├── scaladocs/ │ │ │ ├── xgboost4j/ │ │ │ │ └── index.rst │ │ │ ├── xgboost4j-flink/ │ │ │ │ └── index.rst │ │ │ └── xgboost4j-spark/ │ │ │ └── index.rst │ │ ├── xgboost4j_spark_gpu_tutorial.rst │ │ ├── xgboost4j_spark_tutorial.rst │ │ └── xgboost_spark_migration.rst │ ├── parameter.rst │ ├── prediction.rst │ ├── python/ │ │ ├── .gitignore │ │ ├── callbacks.rst │ │ ├── data_input.rst │ │ ├── index.rst │ │ ├── python_api.rst │ │ ├── python_intro.rst │ │ └── sklearn_estimator.rst │ ├── requirements.txt │ ├── sphinx_util.py │ ├── treemethod.rst │ ├── tutorials/ │ │ ├── advanced_custom_obj.rst │ │ ├── aft_survival_analysis.rst │ │ ├── c_api_tutorial.rst │ │ ├── categorical.rst │ │ ├── custom_metric_obj.rst │ │ ├── dart.rst │ │ ├── dask.rst │ │ ├── external_memory.rst │ │ ├── feature_interaction_constraint.rst │ │ ├── index.rst │ │ ├── input_format.rst │ │ ├── intercept.rst │ │ ├── kubernetes.rst │ │ ├── learning_to_rank.rst │ │ ├── model.rst │ │ ├── monotonic.rst │ │ ├── multioutput.rst │ │ ├── param_tuning.rst │ │ ├── privacy_preserving.rst │ │ ├── ray.rst │ │ ├── rf.rst │ │ ├── saving_model.rst │ │ ├── slicing_model.rst │ │ └── spark_estimator.rst │ └── xgboost_doc.yml ├── include/ │ └── xgboost/ │ ├── base.h │ ├── byteswap.h │ ├── c_api.h │ ├── cache.h │ ├── collective/ │ │ ├── poll_utils.h │ │ ├── result.h │ │ └── socket.h │ ├── context.h │ ├── data.h │ ├── feature_map.h │ ├── gbm.h │ ├── global_config.h │ ├── gradient.h │ ├── host_device_vector.h │ ├── intrusive_ptr.h │ ├── json.h │ ├── json_io.h │ ├── learner.h │ ├── linalg.h │ ├── linear_updater.h │ ├── logging.h │ ├── metric.h │ ├── model.h │ ├── multi_target_tree_model.h │ ├── objective.h │ ├── parameter.h │ ├── predictor.h │ ├── span.h │ ├── string_view.h │ ├── task.h │ ├── tree_model.h │ ├── tree_updater.h │ ├── version_config.h │ └── windefs.h ├── jvm-packages/ │ ├── .gitignore │ ├── CMakeLists.txt │ ├── README.md │ ├── checkstyle-suppressions.xml │ ├── checkstyle.xml │ ├── create_jni.py │ ├── pom.xml │ ├── scalastyle-config.xml │ ├── xgboost4j/ │ │ ├── LICENSE │ │ ├── pom.xml │ │ └── src/ │ │ ├── main/ │ │ │ ├── java/ │ │ │ │ └── ml/ │ │ │ │ └── dmlc/ │ │ │ │ └── xgboost4j/ │ │ │ │ ├── LabeledPoint.java │ │ │ │ └── java/ │ │ │ │ ├── Booster.java │ │ │ │ ├── Column.java │ │ │ │ ├── ColumnBatch.java │ │ │ │ ├── Communicator.java │ │ │ │ ├── ConfigContext.java │ │ │ │ ├── DMatrix.java │ │ │ │ ├── DataBatch.java │ │ │ │ ├── ExternalCheckpointManager.java │ │ │ │ ├── IEvaluation.java │ │ │ │ ├── IObjective.java │ │ │ │ ├── ITracker.java │ │ │ │ ├── NativeLibLoader.java │ │ │ │ ├── RabitTracker.java │ │ │ │ ├── XGBoost.java │ │ │ │ ├── XGBoostError.java │ │ │ │ ├── XGBoostJNI.java │ │ │ │ └── util/ │ │ │ │ ├── BigDenseMatrix.java │ │ │ │ └── UtilUnsafe.java │ │ │ ├── resources/ │ │ │ │ └── xgboost4j-version.properties │ │ │ └── scala/ │ │ │ └── ml/ │ │ │ └── dmlc/ │ │ │ └── xgboost4j/ │ │ │ └── scala/ │ │ │ ├── Booster.scala │ │ │ ├── DMatrix.scala │ │ │ ├── EvalTrait.scala │ │ │ ├── ExternalCheckpointManager.scala │ │ │ ├── ObjectiveTrait.scala │ │ │ └── XGBoost.scala │ │ ├── native/ │ │ │ ├── jvm_utils.h │ │ │ ├── xgboost4j-gpu.cpp │ │ │ ├── xgboost4j-gpu.cu │ │ │ ├── xgboost4j.cpp │ │ │ └── xgboost4j.h │ │ └── test/ │ │ ├── java/ │ │ │ └── ml/ │ │ │ └── dmlc/ │ │ │ └── xgboost4j/ │ │ │ └── java/ │ │ │ ├── ArchDetectionTest.java │ │ │ ├── BoosterImplTest.java │ │ │ ├── ConfigContextTest.java │ │ │ ├── DMatrixTest.java │ │ │ ├── LibraryPathProviderTest.java │ │ │ ├── OsDetectionTest.java │ │ │ └── XGBoostTest.java │ │ └── scala/ │ │ └── ml/ │ │ └── dmlc/ │ │ └── xgboost4j/ │ │ └── scala/ │ │ ├── DMatrixSuite.scala │ │ └── ScalaBoosterImplSuite.scala │ ├── xgboost4j-example/ │ │ ├── LICENSE │ │ ├── README.md │ │ ├── pom.xml │ │ └── src/ │ │ ├── main/ │ │ │ ├── java/ │ │ │ │ └── ml/ │ │ │ │ └── dmlc/ │ │ │ │ └── xgboost4j/ │ │ │ │ └── java/ │ │ │ │ └── example/ │ │ │ │ ├── BasicWalkThrough.java │ │ │ │ ├── BoostFromPrediction.java │ │ │ │ ├── CrossValidation.java │ │ │ │ ├── CustomObjective.java │ │ │ │ ├── EarlyStopping.java │ │ │ │ ├── GeneralizedLinearModel.java │ │ │ │ ├── PredictFirstNtree.java │ │ │ │ ├── PredictLeafIndices.java │ │ │ │ ├── flink/ │ │ │ │ │ └── DistTrainWithFlinkExample.java │ │ │ │ └── util/ │ │ │ │ ├── CustomEval.java │ │ │ │ └── DataLoader.java │ │ │ └── scala/ │ │ │ └── ml/ │ │ │ └── dmlc/ │ │ │ └── xgboost4j/ │ │ │ └── scala/ │ │ │ └── example/ │ │ │ ├── BasicWalkThrough.scala │ │ │ ├── BoostFromPrediction.scala │ │ │ ├── CrossValidation.scala │ │ │ ├── CustomObjective.scala │ │ │ ├── GeneralizedLinearModel.scala │ │ │ ├── PredictFirstNTree.scala │ │ │ ├── PredictLeafIndices.scala │ │ │ ├── flink/ │ │ │ │ └── DistTrainWithFlink.scala │ │ │ ├── spark/ │ │ │ │ ├── SparkMLlibPipeline.scala │ │ │ │ └── SparkTraining.scala │ │ │ └── util/ │ │ │ └── CustomEval.scala │ │ └── test/ │ │ ├── java/ │ │ │ └── ml/ │ │ │ └── dmlc/ │ │ │ └── xgboost4j/ │ │ │ └── java/ │ │ │ └── example/ │ │ │ └── JavaExamplesTest.java │ │ └── scala/ │ │ └── ml/ │ │ └── dmlc/ │ │ └── xgboost4j/ │ │ ├── java/ │ │ │ └── example/ │ │ │ └── flink/ │ │ │ └── DistTrainWithFlinkExampleTest.scala │ │ └── scala/ │ │ └── example/ │ │ ├── ScalaExamplesTest.scala │ │ ├── flink/ │ │ │ └── DistTrainWithFlinkSuite.scala │ │ └── spark/ │ │ └── SparkExamplesTest.scala │ ├── xgboost4j-flink/ │ │ ├── pom.xml │ │ └── src/ │ │ └── main/ │ │ └── java/ │ │ └── ml/ │ │ └── dmlc/ │ │ └── xgboost4j/ │ │ └── java/ │ │ └── flink/ │ │ ├── XGBoost.java │ │ └── XGBoostModel.java │ ├── xgboost4j-spark/ │ │ ├── pom.xml │ │ └── src/ │ │ ├── main/ │ │ │ └── scala/ │ │ │ ├── ml/ │ │ │ │ └── dmlc/ │ │ │ │ └── xgboost4j/ │ │ │ │ └── scala/ │ │ │ │ └── spark/ │ │ │ │ ├── Utils.scala │ │ │ │ ├── XGBoost.scala │ │ │ │ ├── XGBoostClassifier.scala │ │ │ │ ├── XGBoostEstimator.scala │ │ │ │ ├── XGBoostPlugin.scala │ │ │ │ ├── XGBoostRanker.scala │ │ │ │ ├── XGBoostRegressor.scala │ │ │ │ ├── XGBoostTrainingSummary.scala │ │ │ │ ├── package.scala │ │ │ │ └── params/ │ │ │ │ ├── CustomParams.scala │ │ │ │ ├── DartBoosterParams.scala │ │ │ │ ├── GeneralParams.scala │ │ │ │ ├── LearningTaskParams.scala │ │ │ │ ├── ParamMapConversion.scala │ │ │ │ ├── RabitParams.scala │ │ │ │ ├── TreeBoosterParams.scala │ │ │ │ └── XGBoostParams.scala │ │ │ └── org/ │ │ │ └── apache/ │ │ │ └── spark/ │ │ │ └── ml/ │ │ │ └── xgboost/ │ │ │ └── SparkUtils.scala │ │ └── test/ │ │ ├── resources/ │ │ │ ├── log4j.properties │ │ │ ├── model/ │ │ │ │ └── 0.82/ │ │ │ │ └── model/ │ │ │ │ ├── data/ │ │ │ │ │ └── XGBoostClassificationModel │ │ │ │ └── metadata/ │ │ │ │ ├── _SUCCESS │ │ │ │ └── part-00000 │ │ │ └── rank.test.txt │ │ └── scala/ │ │ └── ml/ │ │ └── dmlc/ │ │ └── xgboost4j/ │ │ └── scala/ │ │ └── spark/ │ │ ├── CommunicatorRobustnessSuite.scala │ │ ├── CustomObj.scala │ │ ├── EvalError.scala │ │ ├── PerTest.scala │ │ ├── TmpFolderPerSuite.scala │ │ ├── TrainTestData.scala │ │ ├── XGBoostClassifierSuite.scala │ │ ├── XGBoostEstimatorSuite.scala │ │ ├── XGBoostParamsSuite.scala │ │ ├── XGBoostRankerSuite.scala │ │ ├── XGBoostRegressorSuite.scala │ │ └── XGBoostSuite.scala │ └── xgboost4j-spark-gpu/ │ ├── pom.xml │ └── src/ │ ├── main/ │ │ ├── java/ │ │ │ └── ml/ │ │ │ └── dmlc/ │ │ │ └── xgboost4j/ │ │ │ └── java/ │ │ │ ├── CudfColumn.java │ │ │ ├── CudfColumnBatch.java │ │ │ ├── ExtMemQuantileDMatrix.java │ │ │ └── QuantileDMatrix.java │ │ ├── resources/ │ │ │ └── META-INF/ │ │ │ └── services/ │ │ │ └── ml.dmlc.xgboost4j.scala.spark.XGBoostPlugin │ │ └── scala/ │ │ └── ml/ │ │ └── dmlc/ │ │ └── xgboost4j/ │ │ └── scala/ │ │ ├── ExtMemQuantileDMatrix.scala │ │ ├── QuantileDMatrix.scala │ │ └── spark/ │ │ ├── ExternalMemory.scala │ │ └── GpuXGBoostPlugin.scala │ └── test/ │ ├── java/ │ │ └── ml/ │ │ └── dmlc/ │ │ └── xgboost4j/ │ │ └── java/ │ │ ├── BoosterTest.java │ │ └── DMatrixTest.java │ ├── resources/ │ │ └── log4j.properties │ └── scala/ │ ├── ml/ │ │ └── dmlc/ │ │ └── xgboost4j/ │ │ └── scala/ │ │ ├── ExtMemQuantileDMatrixSuite.scala │ │ ├── QuantileDMatrixSuite.scala │ │ └── spark/ │ │ ├── ExternalMemorySuite.scala │ │ ├── GpuTestSuite.scala │ │ ├── GpuXGBoostPluginSuite.scala │ │ └── TrainTestData.scala │ └── org/ │ └── apache/ │ └── spark/ │ └── GpuTestUtils.scala ├── ops/ │ ├── conda_env/ │ │ ├── aarch64_test.yml │ │ ├── cpp_test.yml │ │ ├── linux_cpu_test.yml │ │ ├── linux_sycl_test.yml │ │ ├── macos_cpu_test.yml │ │ ├── minimal.yml │ │ ├── python_lint.yml │ │ ├── sdist_test.yml │ │ └── win64_test.yml │ ├── docker_run.py │ ├── pipeline/ │ │ ├── audit-cuda-wheel.sh │ │ ├── build-cpu.sh │ │ ├── build-cuda-impl.sh │ │ ├── build-cuda.sh │ │ ├── build-gpu-rpkg.sh │ │ ├── build-jvm-doc.sh │ │ ├── build-jvm-gpu.sh │ │ ├── build-jvm-macos.sh │ │ ├── build-jvm-manylinux2014.sh │ │ ├── build-python-wheels-arm64-impl.sh │ │ ├── build-python-wheels-cpu.sh │ │ ├── build-python-wheels-macos.sh │ │ ├── build-r-docs.sh │ │ ├── build-test-jvm-packages.sh │ │ ├── build-test-sycl.sh │ │ ├── build-variant-wheels-impl.sh │ │ ├── build-variant-wheels.sh │ │ ├── build-win64.ps1 │ │ ├── classify-git-branch.sh │ │ ├── deploy-jvm-packages.sh │ │ ├── enforce-ci.ps1 │ │ ├── enforce-ci.sh │ │ ├── get-docker-registry-details.sh │ │ ├── get-image-tag.sh │ │ ├── login-docker-registry.sh │ │ ├── manage-artifacts.py │ │ ├── nightly-test-cccl-impl.sh │ │ ├── nightly-test-cccl.sh │ │ ├── nightly-test-rmm-impl.sh │ │ ├── nightly-test-rmm.sh │ │ ├── query-latest-cccl.sh │ │ ├── query-latest-rmm.sh │ │ ├── run-clang-tidy.sh │ │ ├── test-c-api-demo.sh │ │ ├── test-cpp-i386.sh │ │ ├── test-freebsd.sh │ │ ├── test-python-macos.sh │ │ ├── test-python-sdist.sh │ │ ├── test-python-wheel.sh │ │ ├── test-python-with-sysprefix.sh │ │ ├── test-win64-gpu.ps1 │ │ ├── trigger-rtd-impl.py │ │ └── trigger-rtd.sh │ └── script/ │ ├── change_scala_version.py │ ├── change_version.py │ ├── changelog.py │ ├── format_wheel_meta.py │ ├── inject_jvm_lib.sh │ ├── lint_cpp.py │ ├── lint_r.R │ ├── pypi_variants.py │ ├── release_artifacts.py │ ├── run_clang_tidy.py │ ├── setup_r_sccache.sh │ ├── test_r_package.py │ ├── test_tidy.cc │ ├── test_utils.py │ ├── type_check_python.py │ └── verify_link.sh ├── plugin/ │ ├── CMakeLists.txt │ ├── README.md │ ├── example/ │ │ ├── README.md │ │ └── custom_obj.cc │ ├── federated/ │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── federated.proto │ │ ├── federated_coll.cc │ │ ├── federated_coll.cu │ │ ├── federated_coll.cuh │ │ ├── federated_coll.h │ │ ├── federated_comm.cc │ │ ├── federated_comm.cu │ │ ├── federated_comm.cuh │ │ ├── federated_comm.h │ │ ├── federated_tracker.cc │ │ └── federated_tracker.h │ ├── sycl/ │ │ ├── README.md │ │ ├── common/ │ │ │ ├── hist_util.cc │ │ │ ├── hist_util.h │ │ │ ├── host_device_vector.cc │ │ │ ├── linalg_op.cc │ │ │ ├── linalg_op.h │ │ │ ├── optional_weight.cc │ │ │ ├── partition_builder.h │ │ │ ├── row_set.h │ │ │ ├── stats.cc │ │ │ └── transform.h │ │ ├── context_helper.cc │ │ ├── context_helper.h │ │ ├── data/ │ │ │ ├── gradient_index.cc │ │ │ └── gradient_index.h │ │ ├── data.h │ │ ├── device_manager.cc │ │ ├── device_manager.h │ │ ├── device_properties.h │ │ ├── predictor/ │ │ │ ├── node.h │ │ │ └── predictor.cc │ │ └── tree/ │ │ ├── expand_entry.h │ │ ├── hist_dispatcher.h │ │ ├── hist_row_adder.h │ │ ├── hist_synchronizer.h │ │ ├── hist_updater.cc │ │ ├── hist_updater.h │ │ ├── param.h │ │ ├── split_evaluator.h │ │ ├── updater_quantile_hist.cc │ │ └── updater_quantile_hist.h │ └── updater_gpu/ │ └── README.md ├── python-package/ │ ├── .gitignore │ ├── README.cpu.rst │ ├── README.dft.rst │ ├── README.rst │ ├── README.stub.rst │ ├── hatch_build.py │ ├── packager/ │ │ ├── __init__.py │ │ ├── build_config.py │ │ ├── nativelib.py │ │ ├── pep517.py │ │ ├── sdist.py │ │ └── util.py │ ├── pyproject.toml │ ├── pyproject.toml.in │ ├── pyproject.toml.stub.in │ └── xgboost/ │ ├── VERSION │ ├── __init__.py │ ├── _c_api.py │ ├── _data_utils.py │ ├── _typing.py │ ├── callback.py │ ├── collective.py │ ├── compat.py │ ├── config.py │ ├── core.py │ ├── dask/ │ │ ├── __init__.py │ │ ├── data.py │ │ └── utils.py │ ├── data.py │ ├── federated.py │ ├── libpath.py │ ├── objective.py │ ├── plotting.py │ ├── py.typed │ ├── sklearn.py │ ├── spark/ │ │ ├── __init__.py │ │ ├── core.py │ │ ├── data.py │ │ ├── estimator.py │ │ ├── params.py │ │ ├── summary.py │ │ └── utils.py │ ├── testing/ │ │ ├── __init__.py │ │ ├── basic_models.py │ │ ├── callbacks.py │ │ ├── collective.py │ │ ├── continuation.py │ │ ├── dask.py │ │ ├── data.py │ │ ├── data_iter.py │ │ ├── federated.py │ │ ├── interaction_constraints.py │ │ ├── intercept.py │ │ ├── metrics.py │ │ ├── monotone_constraints.py │ │ ├── multi_target.py │ │ ├── ordinal.py │ │ ├── params.py │ │ ├── parse_tree.py │ │ ├── plotting.py │ │ ├── predict.py │ │ ├── quantile_dmatrix.py │ │ ├── ranking.py │ │ ├── shared.py │ │ ├── updater.py │ │ ├── utils.py │ │ └── with_skl.py │ ├── tracker.py │ └── training.py ├── src/ │ ├── CMakeLists.txt │ ├── c_api/ │ │ ├── c_api.cc │ │ ├── c_api.cu │ │ ├── c_api_error.cc │ │ ├── c_api_error.h │ │ ├── c_api_utils.h │ │ └── coll_c_api.cc │ ├── collective/ │ │ ├── aggregator.cuh │ │ ├── aggregator.h │ │ ├── allgather.cc │ │ ├── allgather.h │ │ ├── allreduce.cc │ │ ├── allreduce.h │ │ ├── broadcast.cc │ │ ├── broadcast.h │ │ ├── coll.cc │ │ ├── coll.cu │ │ ├── coll.cuh │ │ ├── coll.h │ │ ├── comm.cc │ │ ├── comm.cu │ │ ├── comm.cuh │ │ ├── comm.h │ │ ├── comm_group.cc │ │ ├── comm_group.h │ │ ├── communicator-inl.h │ │ ├── in_memory_communicator.h │ │ ├── in_memory_handler.cc │ │ ├── in_memory_handler.h │ │ ├── loop.cc │ │ ├── loop.h │ │ ├── nccl_stub.cc │ │ ├── nccl_stub.h │ │ ├── protocol.h │ │ ├── result.cc │ │ ├── socket.cc │ │ ├── topo.h │ │ ├── tracker.cc │ │ └── tracker.h │ ├── common/ │ │ ├── algorithm.cuh │ │ ├── algorithm.h │ │ ├── api_entry.h │ │ ├── base64.h │ │ ├── bitfield.h │ │ ├── cache_manager.cc │ │ ├── cache_manager.h │ │ ├── categorical.h │ │ ├── charconv.cc │ │ ├── charconv.h │ │ ├── column_matrix.cc │ │ ├── column_matrix.h │ │ ├── common.cc │ │ ├── common.cu │ │ ├── common.h │ │ ├── compressed_iterator.h │ │ ├── cuda_context.cuh │ │ ├── cuda_dr_utils.cc │ │ ├── cuda_dr_utils.h │ │ ├── cuda_pinned_allocator.cu │ │ ├── cuda_pinned_allocator.h │ │ ├── cuda_rt_utils.cc │ │ ├── cuda_rt_utils.h │ │ ├── cuda_stream.h │ │ ├── cuda_stream_pool.h │ │ ├── deterministic.cuh │ │ ├── device_compression.cu │ │ ├── device_compression.cuh │ │ ├── device_compression.h │ │ ├── device_debug.cuh │ │ ├── device_helpers.cu │ │ ├── device_helpers.cuh │ │ ├── device_vector.cu │ │ ├── device_vector.cuh │ │ ├── error_msg.cc │ │ ├── error_msg.h │ │ ├── expectile_loss_utils.cc │ │ ├── expectile_loss_utils.h │ │ ├── group_data.h │ │ ├── hist_util.cc │ │ ├── hist_util.cu │ │ ├── hist_util.cuh │ │ ├── hist_util.h │ │ ├── host_device_vector.cc │ │ ├── host_device_vector.cu │ │ ├── io.cc │ │ ├── io.h │ │ ├── json.cc │ │ ├── json_utils.h │ │ ├── linalg_op.cc │ │ ├── linalg_op.cu │ │ ├── linalg_op.cuh │ │ ├── linalg_op.h │ │ ├── math.h │ │ ├── numa_topo.cc │ │ ├── numa_topo.h │ │ ├── numeric.cc │ │ ├── numeric.cu │ │ ├── numeric.h │ │ ├── nvtx_utils.h │ │ ├── observer.h │ │ ├── optional_weight.cc │ │ ├── optional_weight.cu │ │ ├── optional_weight.h │ │ ├── param_array.cc │ │ ├── param_array.h │ │ ├── partition_builder.h │ │ ├── probability_distribution.h │ │ ├── pseudo_huber.cc │ │ ├── pseudo_huber.h │ │ ├── quantile.cc │ │ ├── quantile.cu │ │ ├── quantile.cuh │ │ ├── quantile.h │ │ ├── quantile_loss_utils.cc │ │ ├── quantile_loss_utils.h │ │ ├── random.cc │ │ ├── random.cu │ │ ├── random.h │ │ ├── ranking_utils.cc │ │ ├── ranking_utils.cu │ │ ├── ranking_utils.cuh │ │ ├── ranking_utils.h │ │ ├── ref_resource_view.cuh │ │ ├── ref_resource_view.h │ │ ├── resource.cu │ │ ├── resource.cuh │ │ ├── row_set.h │ │ ├── stats.cc │ │ ├── stats.cu │ │ ├── stats.cuh │ │ ├── stats.h │ │ ├── survival_util.cc │ │ ├── survival_util.h │ │ ├── threading_utils.cc │ │ ├── threading_utils.cuh │ │ ├── threading_utils.h │ │ ├── threadpool.h │ │ ├── timer.cc │ │ ├── timer.h │ │ ├── transform.h │ │ ├── transform_iterator.h │ │ ├── type.h │ │ ├── utils.h │ │ ├── version.cc │ │ └── version.h │ ├── context.cc │ ├── context.cu │ ├── data/ │ │ ├── adapter.cc │ │ ├── adapter.h │ │ ├── array_interface.cc │ │ ├── array_interface.cu │ │ ├── array_interface.h │ │ ├── batch_utils.cc │ │ ├── batch_utils.h │ │ ├── cat_container.cc │ │ ├── cat_container.cu │ │ ├── cat_container.cuh │ │ ├── cat_container.h │ │ ├── columnar.h │ │ ├── data.cc │ │ ├── data.cu │ │ ├── device_adapter.cu │ │ ├── device_adapter.cuh │ │ ├── ellpack_page.cc │ │ ├── ellpack_page.cu │ │ ├── ellpack_page.cuh │ │ ├── ellpack_page.h │ │ ├── ellpack_page_raw_format.cu │ │ ├── ellpack_page_raw_format.h │ │ ├── ellpack_page_source.cu │ │ ├── ellpack_page_source.h │ │ ├── entry.h │ │ ├── extmem_quantile_dmatrix.cc │ │ ├── extmem_quantile_dmatrix.cu │ │ ├── extmem_quantile_dmatrix.h │ │ ├── file_iterator.cc │ │ ├── file_iterator.h │ │ ├── gradient_index.cc │ │ ├── gradient_index.cu │ │ ├── gradient_index.h │ │ ├── gradient_index_format.cc │ │ ├── gradient_index_format.h │ │ ├── gradient_index_page_source.cc │ │ ├── gradient_index_page_source.h │ │ ├── iterative_dmatrix.cc │ │ ├── iterative_dmatrix.cu │ │ ├── iterative_dmatrix.h │ │ ├── metainfo.cc │ │ ├── metainfo.cu │ │ ├── metainfo.h │ │ ├── proxy_dmatrix.cc │ │ ├── proxy_dmatrix.cu │ │ ├── proxy_dmatrix.cuh │ │ ├── proxy_dmatrix.h │ │ ├── quantile_dmatrix.cc │ │ ├── quantile_dmatrix.cu │ │ ├── quantile_dmatrix.h │ │ ├── simple_batch_iterator.h │ │ ├── simple_dmatrix.cc │ │ ├── simple_dmatrix.cu │ │ ├── simple_dmatrix.cuh │ │ ├── simple_dmatrix.h │ │ ├── sparse_page_dmatrix.cc │ │ ├── sparse_page_dmatrix.cu │ │ ├── sparse_page_dmatrix.h │ │ ├── sparse_page_raw_format.cc │ │ ├── sparse_page_source.cc │ │ ├── sparse_page_source.cu │ │ ├── sparse_page_source.h │ │ └── sparse_page_writer.h │ ├── encoder/ │ │ ├── ordinal.cuh │ │ ├── ordinal.h │ │ └── types.h │ ├── gbm/ │ │ ├── gblinear.cc │ │ ├── gblinear_model.cc │ │ ├── gblinear_model.h │ │ ├── gbm.cc │ │ ├── gbtree.cc │ │ ├── gbtree.cu │ │ ├── gbtree.h │ │ ├── gbtree_model.cc │ │ └── gbtree_model.h │ ├── global_config.cc │ ├── learner.cc │ ├── linear/ │ │ ├── coordinate_common.h │ │ ├── linear_updater.cc │ │ ├── param.h │ │ ├── updater_coordinate.cc │ │ ├── updater_gpu_coordinate.cu │ │ └── updater_shotgun.cc │ ├── logging.cc │ ├── metric/ │ │ ├── auc.cc │ │ ├── auc.cu │ │ ├── auc.h │ │ ├── elementwise_metric.cc │ │ ├── elementwise_metric.cu │ │ ├── metric.cc │ │ ├── metric_common.h │ │ ├── multiclass_metric.cc │ │ ├── multiclass_metric.cu │ │ ├── rank_metric.cc │ │ ├── rank_metric.cu │ │ ├── rank_metric.h │ │ ├── survival_metric.cc │ │ └── survival_metric.cu │ ├── objective/ │ │ ├── adaptive.cc │ │ ├── adaptive.cu │ │ ├── adaptive.h │ │ ├── aft_obj.cc │ │ ├── aft_obj.cu │ │ ├── hinge.cc │ │ ├── hinge.cu │ │ ├── init_estimation.cc │ │ ├── init_estimation.h │ │ ├── lambdarank_obj.cc │ │ ├── lambdarank_obj.cu │ │ ├── lambdarank_obj.cuh │ │ ├── lambdarank_obj.h │ │ ├── multiclass_obj.cc │ │ ├── multiclass_obj.cu │ │ ├── multiclass_param.h │ │ ├── objective.cc │ │ ├── quantile_obj.cc │ │ ├── quantile_obj.cu │ │ ├── regression_loss.h │ │ ├── regression_obj.cc │ │ ├── regression_obj.cu │ │ └── regression_param.h │ ├── predictor/ │ │ ├── array_tree_layout.h │ │ ├── cpu_predictor.cc │ │ ├── data_accessor.h │ │ ├── gbtree_view.h │ │ ├── gpu_data_accessor.cuh │ │ ├── gpu_predictor.cu │ │ ├── interpretability/ │ │ │ ├── shap.cc │ │ │ ├── shap.cu │ │ │ └── shap.h │ │ ├── predict_fn.h │ │ ├── predictor.cc │ │ ├── predictor.cu │ │ ├── treeshap.cc │ │ ├── treeshap.h │ │ └── utils.h │ └── tree/ │ ├── common_row_partitioner.h │ ├── constraints.cc │ ├── constraints.cu │ ├── constraints.cuh │ ├── constraints.h │ ├── driver.h │ ├── fit_stump.cc │ ├── fit_stump.cu │ ├── fit_stump.h │ ├── gpu_hist/ │ │ ├── evaluate_splits.cu │ │ ├── evaluate_splits.cuh │ │ ├── evaluator.cu │ │ ├── expand_entry.cu │ │ ├── expand_entry.cuh │ │ ├── feature_groups.cu │ │ ├── feature_groups.cuh │ │ ├── histogram.cu │ │ ├── histogram.cuh │ │ ├── leaf_sum.cu │ │ ├── leaf_sum.cuh │ │ ├── multi_evaluate_splits.cu │ │ ├── multi_evaluate_splits.cuh │ │ ├── quantiser.cu │ │ ├── quantiser.cuh │ │ ├── row_partitioner.cu │ │ ├── row_partitioner.cuh │ │ ├── sampler.cu │ │ └── sampler.cuh │ ├── hist/ │ │ ├── evaluate_splits.h │ │ ├── expand_entry.h │ │ ├── hist_cache.h │ │ ├── hist_param.cc │ │ ├── hist_param.h │ │ ├── histogram.cc │ │ ├── histogram.h │ │ ├── sampler.cc │ │ └── sampler.h │ ├── io_utils.h │ ├── multi_target_tree_model.cc │ ├── param.cc │ ├── param.h │ ├── sample_position.h │ ├── split_evaluator.h │ ├── tree_model.cc │ ├── tree_updater.cc │ ├── tree_view.cc │ ├── tree_view.h │ ├── updater_approx.cc │ ├── updater_colmaker.cc │ ├── updater_gpu_common.cuh │ ├── updater_gpu_hist.cu │ ├── updater_gpu_hist.cuh │ ├── updater_prune.cc │ ├── updater_quantile_hist.cc │ ├── updater_refresh.cc │ └── updater_sync.cc └── tests/ ├── README.md ├── cpp/ │ ├── CMakeLists.txt │ ├── c_api/ │ │ └── test_c_api.cc │ ├── categorical_helpers.h │ ├── collective/ │ │ ├── test_allgather.cc │ │ ├── test_allgather.cu │ │ ├── test_allreduce.cc │ │ ├── test_allreduce.cu │ │ ├── test_broadcast.cc │ │ ├── test_coll_c_api.cc │ │ ├── test_comm.cc │ │ ├── test_comm_group.cc │ │ ├── test_loop.cc │ │ ├── test_result.cc │ │ ├── test_socket.cc │ │ ├── test_tracker.cc │ │ ├── test_worker.cuh │ │ └── test_worker.h │ ├── common/ │ │ ├── test_algorithm.cc │ │ ├── test_algorithm.cu │ │ ├── test_bitfield.cc │ │ ├── test_bitfield.cu │ │ ├── test_categorical.cc │ │ ├── test_charconv.cc │ │ ├── test_column_matrix.cc │ │ ├── test_common.cc │ │ ├── test_compressed_iterator.cc │ │ ├── test_cuda_dr_utils.cc │ │ ├── test_cuda_host_allocator.cu │ │ ├── test_cuda_rt_utils.cu │ │ ├── test_device_compression.cu │ │ ├── test_device_helpers.cu │ │ ├── test_device_vector.cu │ │ ├── test_gpu_compressed_iterator.cu │ │ ├── test_group_data.cc │ │ ├── test_hist_util.cc │ │ ├── test_hist_util.cu │ │ ├── test_hist_util.h │ │ ├── test_host_device_vector.cu │ │ ├── test_intrusive_ptr.cc │ │ ├── test_io.cc │ │ ├── test_json.cc │ │ ├── test_linalg.cc │ │ ├── test_linalg.cu │ │ ├── test_linalg.h │ │ ├── test_math.cc │ │ ├── test_monitor.cc │ │ ├── test_numa_topo.cc │ │ ├── test_numeric.cc │ │ ├── test_optional_weight.cc │ │ ├── test_param_array.cc │ │ ├── test_parameter.cc │ │ ├── test_partition_builder.cc │ │ ├── test_probability_distribution.cc │ │ ├── test_quantile.cc │ │ ├── test_quantile.cu │ │ ├── test_quantile.h │ │ ├── test_quantile_utils.cc │ │ ├── test_random.cc │ │ ├── test_ranking_utils.cc │ │ ├── test_ranking_utils.cu │ │ ├── test_ranking_utils.h │ │ ├── test_ref_resource_view.cc │ │ ├── test_ref_resource_view.cu │ │ ├── test_span.cc │ │ ├── test_span.cu │ │ ├── test_span.h │ │ ├── test_stats.cc │ │ ├── test_stats.cu │ │ ├── test_string_view.cc │ │ ├── test_survival_util.cc │ │ ├── test_threading_utils.cc │ │ ├── test_threading_utils.cu │ │ ├── test_threadpool.cc │ │ ├── test_transform_iterator.cc │ │ ├── test_transform_range.cc │ │ ├── test_transform_range.cu │ │ └── test_version.cc │ ├── data/ │ │ ├── test_adapter.cc │ │ ├── test_array_interface.cc │ │ ├── test_array_interface.cu │ │ ├── test_array_interface.h │ │ ├── test_batch_utils.cu │ │ ├── test_cat_container.cc │ │ ├── test_cat_container.cu │ │ ├── test_cat_container.h │ │ ├── test_data.cc │ │ ├── test_device_adapter.cu │ │ ├── test_ellpack_page.cu │ │ ├── test_ellpack_page_raw_format.cu │ │ ├── test_extmem_quantile_dmatrix.cc │ │ ├── test_extmem_quantile_dmatrix.cu │ │ ├── test_extmem_quantile_dmatrix.h │ │ ├── test_file_iterator.cc │ │ ├── test_gradient_index.cc │ │ ├── test_gradient_index_page_raw_format.cc │ │ ├── test_iterative_dmatrix.cc │ │ ├── test_iterative_dmatrix.cu │ │ ├── test_iterative_dmatrix.h │ │ ├── test_metainfo.cc │ │ ├── test_metainfo.cu │ │ ├── test_metainfo.h │ │ ├── test_proxy_dmatrix.cc │ │ ├── test_proxy_dmatrix.cu │ │ ├── test_simple_dmatrix.cc │ │ ├── test_simple_dmatrix.cu │ │ ├── test_sparse_page_dmatrix.cc │ │ ├── test_sparse_page_dmatrix.cu │ │ └── test_sparse_page_raw_format.cc │ ├── encoder/ │ │ ├── df_mock.cuh │ │ ├── df_mock.h │ │ ├── test_ordinal.cc │ │ ├── test_ordinal.cu │ │ └── test_ordinal.h │ ├── filesystem.cc │ ├── filesystem.h │ ├── gbm/ │ │ ├── test_gblinear.cc │ │ ├── test_gblinear.cu │ │ ├── test_gbtree.cc │ │ └── test_gbtree.cu │ ├── helpers.cc │ ├── helpers.cu │ ├── helpers.h │ ├── histogram_helpers.cu │ ├── histogram_helpers.h │ ├── linear/ │ │ ├── test_json_io.h │ │ ├── test_linear.cc │ │ └── test_linear.cu │ ├── metric/ │ │ ├── test_auc.h │ │ ├── test_distributed_metric.cc │ │ ├── test_elementwise_metric.h │ │ ├── test_metric.cc │ │ ├── test_multiclass_metric.h │ │ ├── test_rank_metric.cc │ │ ├── test_rank_metric.h │ │ ├── test_survival_metric.cc │ │ ├── test_survival_metric.cu │ │ └── test_survival_metric.h │ ├── objective/ │ │ ├── test_aft_obj.cc │ │ ├── test_aft_obj.cu │ │ ├── test_aft_obj.h │ │ ├── test_aft_obj_cpu.cc │ │ ├── test_hinge.cc │ │ ├── test_hinge.cu │ │ ├── test_hinge.h │ │ ├── test_hinge_cpu.cc │ │ ├── test_lambdarank_obj.cc │ │ ├── test_lambdarank_obj.cu │ │ ├── test_lambdarank_obj.h │ │ ├── test_multiclass_obj.cc │ │ ├── test_multiclass_obj.h │ │ ├── test_multiclass_obj_cpu.cc │ │ ├── test_multiclass_obj_gpu.cu │ │ ├── test_objective.cc │ │ ├── test_objective_helpers.h │ │ ├── test_quantile_obj.cc │ │ ├── test_quantile_obj.h │ │ ├── test_quantile_obj_cpu.cc │ │ ├── test_quantile_obj_gpu.cu │ │ ├── test_regression_obj.cc │ │ ├── test_regression_obj.h │ │ ├── test_regression_obj_cpu.cc │ │ └── test_regression_obj_gpu.cu │ ├── objective_helpers.cc │ ├── objective_helpers.h │ ├── plugin/ │ │ ├── federated/ │ │ │ ├── CMakeLists.txt │ │ │ ├── test_federated_coll.cc │ │ │ ├── test_federated_coll.cu │ │ │ ├── test_federated_comm.cc │ │ │ ├── test_federated_comm_group.cc │ │ │ ├── test_federated_comm_group.cu │ │ │ ├── test_federated_data.cc │ │ │ ├── test_federated_learner.cc │ │ │ ├── test_federated_tracker.cc │ │ │ └── test_worker.h │ │ ├── sycl_helpers.h │ │ ├── test_example_objective.cc │ │ ├── test_sycl_aft_obj.cc │ │ ├── test_sycl_ghist_builder.cc │ │ ├── test_sycl_gradient_index.cc │ │ ├── test_sycl_hinge.cc │ │ ├── test_sycl_hist_updater.cc │ │ ├── test_sycl_host_device_vector.cc │ │ ├── test_sycl_lambdarank_obj.cc │ │ ├── test_sycl_linalg.cc │ │ ├── test_sycl_multiclass_obj.cc │ │ ├── test_sycl_partition_builder.cc │ │ ├── test_sycl_prediction_cache.cc │ │ ├── test_sycl_predictor.cc │ │ ├── test_sycl_quantile_hist_builder.cc │ │ ├── test_sycl_quantile_obj.cc │ │ ├── test_sycl_regression_obj.cc │ │ ├── test_sycl_row_set_collection.cc │ │ ├── test_sycl_split_evaluator.cc │ │ └── test_sycl_transform_range.cc │ ├── predictor/ │ │ ├── test_cpu_predictor.cc │ │ ├── test_gpu_predictor.cu │ │ ├── test_predictor.cc │ │ ├── test_predictor.h │ │ ├── test_shap.cc │ │ ├── test_shap.cu │ │ └── test_shap.h │ ├── test_cache.cc │ ├── test_context.cc │ ├── test_context.cu │ ├── test_global_config.cc │ ├── test_helpers.cc │ ├── test_histogram_helpers.cu │ ├── test_learner.cc │ ├── test_learner.cu │ ├── test_logging.cc │ ├── test_main.cc │ ├── test_multi_target.cc │ ├── test_serialization.cc │ ├── test_serialization.h │ └── tree/ │ ├── gpu_hist/ │ │ ├── dummy_quantizer.cuh │ │ ├── test_driver.cu │ │ ├── test_evaluate_splits.cu │ │ ├── test_expand_entry.cu │ │ ├── test_histogram.cu │ │ ├── test_leaf_sum.cu │ │ ├── test_multi_evaluate_splits.cu │ │ ├── test_multi_histogram.cu │ │ ├── test_row_partitioner.cu │ │ └── test_sampler.cu │ ├── hist/ │ │ ├── test_evaluate_splits.cc │ │ ├── test_expand_entry.cc │ │ ├── test_histogram.cc │ │ └── test_sampler.cc │ ├── test_approx.cc │ ├── test_column_split.cc │ ├── test_column_split.h │ ├── test_common_partitioner.cc │ ├── test_constraints.cc │ ├── test_constraints.cu │ ├── test_evaluate_splits.h │ ├── test_fit_stump.cc │ ├── test_gpu_approx.cu │ ├── test_gpu_hist.cu │ ├── test_multi_target_tree_model.cc │ ├── test_multi_target_tree_model.h │ ├── test_node_partition.cc │ ├── test_param.cc │ ├── test_partitioner.h │ ├── test_prediction_cache.cc │ ├── test_prediction_cache.h │ ├── test_prune.cc │ ├── test_quantile_hist.cc │ ├── test_refresh.cc │ ├── test_regen.cc │ ├── test_sampler.h │ ├── test_tree_model.cc │ ├── test_tree_policy.cc │ └── test_tree_stat.cc ├── cross-platform/ │ └── test_cross_platform_model.py ├── pytest.ini ├── python/ │ ├── generate_models.py │ ├── test_basic.py │ ├── test_basic_models.py │ ├── test_callback.py │ ├── test_collective.py │ ├── test_config.py │ ├── test_data_iterator.py │ ├── test_demos.py │ ├── test_dmatrix.py │ ├── test_early_stopping.py │ ├── test_eval_metrics.py │ ├── test_interaction_constraints.py │ ├── test_intercept.py │ ├── test_linear.py │ ├── test_model_compatibility.py │ ├── test_model_io.py │ ├── test_monotone_constraints.py │ ├── test_multi_target.py │ ├── test_openmp.py │ ├── test_ordinal.py │ ├── test_parse_tree.py │ ├── test_pickling.py │ ├── test_plotting.py │ ├── test_predict.py │ ├── test_quantile_dmatrix.py │ ├── test_ranking.py │ ├── test_shap.py │ ├── test_survival.py │ ├── test_tracker.py │ ├── test_training_continuation.py │ ├── test_tree_regularization.py │ ├── test_updaters.py │ ├── test_with_arrow.py │ ├── test_with_modin.py │ ├── test_with_pandas.py │ ├── test_with_polars.py │ ├── test_with_scipy.py │ ├── test_with_shap.py │ ├── test_with_sklearn.py │ └── with_omp_limit.py ├── python-gpu/ │ ├── conftest.py │ ├── load_pickle.py │ ├── test_device_quantile_dmatrix.py │ ├── test_from_cudf.py │ ├── test_from_cupy.py │ ├── test_gpu_basic_models.py │ ├── test_gpu_callbacks.py │ ├── test_gpu_data_iterator.py │ ├── test_gpu_demos.py │ ├── test_gpu_eval_metrics.py │ ├── test_gpu_interaction_constraints.py │ ├── test_gpu_intercept.py │ ├── test_gpu_linear.py │ ├── test_gpu_multi_target.py │ ├── test_gpu_ordinal.py │ ├── test_gpu_parse_tree.py │ ├── test_gpu_pickling.py │ ├── test_gpu_plotting.py │ ├── test_gpu_prediction.py │ ├── test_gpu_ranking.py │ ├── test_gpu_training_continuation.py │ ├── test_gpu_updaters.py │ ├── test_gpu_with_sklearn.py │ ├── test_large_input.py │ └── test_monotonic_constraints.py ├── python-sycl/ │ ├── test_sycl_prediction.py │ ├── test_sycl_simple_dask.py │ ├── test_sycl_training_continuation.py │ ├── test_sycl_updaters.py │ └── test_sycl_with_sklearn.py └── test_distributed/ ├── __init__.py ├── test_federated/ │ └── test_federated.py ├── test_gpu_federated/ │ └── test_gpu_federated.py ├── test_gpu_with_dask/ │ ├── __init__.py │ ├── conftest.py │ ├── test_gpu_demos.py │ ├── test_gpu_external_memory.py │ ├── test_gpu_ranking.py │ └── test_gpu_with_dask.py ├── test_with_dask/ │ ├── __init__.py │ ├── conftest.py │ ├── test_demos.py │ ├── test_external_memory.py │ ├── test_ranking.py │ └── test_with_dask.py └── test_with_spark/ ├── __init__.py ├── discover_gpu.sh ├── test_data.py ├── test_spark.py └── utils.py