├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── scripts │ ├── build_source.sh │ ├── build_wheels.sh │ ├── test_source.sh │ └── test_wheels.sh └── workflows │ └── build_wheels.yml ├── .gitignore ├── .readthedocs.yaml ├── LICENSE ├── MANIFEST.in ├── NOTICE ├── README.md ├── changelog.md ├── docs ├── Makefile ├── conf.py ├── distributed_learning.md ├── examples.md ├── features.md ├── figures │ ├── quick_start_firstexample.png │ └── quick_start_tunedexample.png ├── function_reference.rst ├── index.rst ├── installation.md ├── make.bat ├── parameters_sklearn.md ├── parameters_torch.md ├── quick_start.md ├── requirements.txt └── support.md ├── examples ├── README.md ├── sklearn │ ├── README.md │ ├── datasets.py │ ├── example01_housing_cpu.py │ ├── example04_housing_validation.py │ ├── example05_housing_vsngboost.py │ ├── example06_trainingtimevslightgbm.py │ ├── example07_hyperparamoptimization.py │ ├── example09_housing_featimportance.py │ ├── example11_housing_saveandload.py │ ├── example12_housing_checkpointing.py │ ├── example13_monotone_constraints.py │ └── example14_probregression.py ├── torch │ ├── README.md │ ├── datasets.py │ ├── example01_housing_cpu.ipynb │ ├── example01_housing_cpu.py │ ├── example02_housing_gpu.ipynb │ ├── example02_housing_gpu.py │ ├── example04_housing_validation.py │ ├── example05_housing_vsngboost.py │ ├── example06_trainingtimevslightgbm.py │ ├── example07_optimizeddistribution.py │ ├── example08_housing_autodiff.py │ ├── example09_housing_featimportance.py │ ├── example09a_housing_featimportance_shaps.py │ ├── example10_covidhospitaladmissions.py │ ├── example11_housing_saveandload.py │ ├── example12_housing_checkpointing.py │ └── example15_monotone_constraints.py └── torch_dist │ ├── README.md │ ├── dist_template.py │ ├── example13_housing_dist.py │ └── example14_higgs_dist.py ├── paper ├── datasets │ ├── __init__.py │ └── datasets.py ├── experiments │ ├── 01_uci_benchmark │ │ ├── 01_uci_benchmark_lightgbm.py │ │ ├── 01_uci_benchmark_ngboost.py │ │ └── 01_uci_benchmark_pgbm.py │ ├── 01a_rho_variation │ │ ├── 01a_rho_variation.py │ │ └── 01a_rho_variation_leaves_variation.py │ ├── 01b_posterior_distribution │ │ └── 01b_posterior_distribution.py │ └── 02_hierarchical_time_series │ │ ├── 01_preprocessing.py │ │ ├── 02a_training_lgbm_mse.py │ │ ├── 02b_training_pgbm_mse.py │ │ ├── 02c_training_pgbm_wmse.py │ │ ├── 02d_training_ngboost_mse.py │ │ └── 03_evaluation.py └── plots.py ├── pgbm ├── __init__.py ├── sklearn │ ├── __init__.py │ ├── _binning.pyx │ ├── _bitset.pyx │ ├── _gradient_boosting.pyx │ ├── _predictor.pyx │ ├── binning.py │ ├── common.pyx │ ├── distributions.py │ ├── gradient_boosting.py │ ├── grower.py │ ├── histogram.pyx │ ├── predictor.py │ ├── splitting.pyx │ ├── tests │ │ ├── __init__.py │ │ ├── test_binning.py │ │ ├── test_bitset.py │ │ ├── test_compare_lightgbm.py │ │ ├── test_estimator.py │ │ ├── test_gradient_boosting.py │ │ ├── test_grower.py │ │ ├── test_histogram.py │ │ ├── test_monotonic_contraints.py │ │ ├── test_predictor.py │ │ ├── test_splitting.py │ │ └── test_warm_start.py │ └── utils.pyx └── torch │ ├── __init__.py │ ├── pgbm.py │ ├── pgbm_dist.py │ ├── splitgain_cpu.cpp │ ├── splitgain_cuda.cpp │ ├── splitgain_kernel.cu │ └── tests │ ├── __init__.py │ └── test_estimator.py ├── pyproject.toml └── setup.py /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/scripts/build_source.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.github/scripts/build_source.sh -------------------------------------------------------------------------------- /.github/scripts/build_wheels.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.github/scripts/build_wheels.sh -------------------------------------------------------------------------------- /.github/scripts/test_source.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.github/scripts/test_source.sh -------------------------------------------------------------------------------- /.github/scripts/test_wheels.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.github/scripts/test_wheels.sh -------------------------------------------------------------------------------- /.github/workflows/build_wheels.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.github/workflows/build_wheels.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.gitignore -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/README.md -------------------------------------------------------------------------------- /changelog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/changelog.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/distributed_learning.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/distributed_learning.md -------------------------------------------------------------------------------- /docs/examples.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/examples.md -------------------------------------------------------------------------------- /docs/features.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/features.md -------------------------------------------------------------------------------- /docs/figures/quick_start_firstexample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/figures/quick_start_firstexample.png -------------------------------------------------------------------------------- /docs/figures/quick_start_tunedexample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/figures/quick_start_tunedexample.png -------------------------------------------------------------------------------- /docs/function_reference.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/function_reference.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/installation.md -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/parameters_sklearn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/parameters_sklearn.md -------------------------------------------------------------------------------- /docs/parameters_torch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/parameters_torch.md -------------------------------------------------------------------------------- /docs/quick_start.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/quick_start.md -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/support.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/docs/support.md -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/sklearn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/README.md -------------------------------------------------------------------------------- /examples/sklearn/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/datasets.py -------------------------------------------------------------------------------- /examples/sklearn/example01_housing_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example01_housing_cpu.py -------------------------------------------------------------------------------- /examples/sklearn/example04_housing_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example04_housing_validation.py -------------------------------------------------------------------------------- /examples/sklearn/example05_housing_vsngboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example05_housing_vsngboost.py -------------------------------------------------------------------------------- /examples/sklearn/example06_trainingtimevslightgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example06_trainingtimevslightgbm.py -------------------------------------------------------------------------------- /examples/sklearn/example07_hyperparamoptimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example07_hyperparamoptimization.py -------------------------------------------------------------------------------- /examples/sklearn/example09_housing_featimportance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example09_housing_featimportance.py -------------------------------------------------------------------------------- /examples/sklearn/example11_housing_saveandload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example11_housing_saveandload.py -------------------------------------------------------------------------------- /examples/sklearn/example12_housing_checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example12_housing_checkpointing.py -------------------------------------------------------------------------------- /examples/sklearn/example13_monotone_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example13_monotone_constraints.py -------------------------------------------------------------------------------- /examples/sklearn/example14_probregression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/sklearn/example14_probregression.py -------------------------------------------------------------------------------- /examples/torch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/README.md -------------------------------------------------------------------------------- /examples/torch/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/datasets.py -------------------------------------------------------------------------------- /examples/torch/example01_housing_cpu.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example01_housing_cpu.ipynb -------------------------------------------------------------------------------- /examples/torch/example01_housing_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example01_housing_cpu.py -------------------------------------------------------------------------------- /examples/torch/example02_housing_gpu.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example02_housing_gpu.ipynb -------------------------------------------------------------------------------- /examples/torch/example02_housing_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example02_housing_gpu.py -------------------------------------------------------------------------------- /examples/torch/example04_housing_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example04_housing_validation.py -------------------------------------------------------------------------------- /examples/torch/example05_housing_vsngboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example05_housing_vsngboost.py -------------------------------------------------------------------------------- /examples/torch/example06_trainingtimevslightgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example06_trainingtimevslightgbm.py -------------------------------------------------------------------------------- /examples/torch/example07_optimizeddistribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example07_optimizeddistribution.py -------------------------------------------------------------------------------- /examples/torch/example08_housing_autodiff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example08_housing_autodiff.py -------------------------------------------------------------------------------- /examples/torch/example09_housing_featimportance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example09_housing_featimportance.py -------------------------------------------------------------------------------- /examples/torch/example09a_housing_featimportance_shaps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example09a_housing_featimportance_shaps.py -------------------------------------------------------------------------------- /examples/torch/example10_covidhospitaladmissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example10_covidhospitaladmissions.py -------------------------------------------------------------------------------- /examples/torch/example11_housing_saveandload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example11_housing_saveandload.py -------------------------------------------------------------------------------- /examples/torch/example12_housing_checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example12_housing_checkpointing.py -------------------------------------------------------------------------------- /examples/torch/example15_monotone_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch/example15_monotone_constraints.py -------------------------------------------------------------------------------- /examples/torch_dist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch_dist/README.md -------------------------------------------------------------------------------- /examples/torch_dist/dist_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch_dist/dist_template.py -------------------------------------------------------------------------------- /examples/torch_dist/example13_housing_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch_dist/example13_housing_dist.py -------------------------------------------------------------------------------- /examples/torch_dist/example14_higgs_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/examples/torch_dist/example14_higgs_dist.py -------------------------------------------------------------------------------- /paper/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/datasets/__init__.py -------------------------------------------------------------------------------- /paper/datasets/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/datasets/datasets.py -------------------------------------------------------------------------------- /paper/experiments/01_uci_benchmark/01_uci_benchmark_lightgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/01_uci_benchmark/01_uci_benchmark_lightgbm.py -------------------------------------------------------------------------------- /paper/experiments/01_uci_benchmark/01_uci_benchmark_ngboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/01_uci_benchmark/01_uci_benchmark_ngboost.py -------------------------------------------------------------------------------- /paper/experiments/01_uci_benchmark/01_uci_benchmark_pgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/01_uci_benchmark/01_uci_benchmark_pgbm.py -------------------------------------------------------------------------------- /paper/experiments/01a_rho_variation/01a_rho_variation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/01a_rho_variation/01a_rho_variation.py -------------------------------------------------------------------------------- /paper/experiments/01a_rho_variation/01a_rho_variation_leaves_variation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/01a_rho_variation/01a_rho_variation_leaves_variation.py -------------------------------------------------------------------------------- /paper/experiments/01b_posterior_distribution/01b_posterior_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/01b_posterior_distribution/01b_posterior_distribution.py -------------------------------------------------------------------------------- /paper/experiments/02_hierarchical_time_series/01_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/02_hierarchical_time_series/01_preprocessing.py -------------------------------------------------------------------------------- /paper/experiments/02_hierarchical_time_series/02a_training_lgbm_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/02_hierarchical_time_series/02a_training_lgbm_mse.py -------------------------------------------------------------------------------- /paper/experiments/02_hierarchical_time_series/02b_training_pgbm_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/02_hierarchical_time_series/02b_training_pgbm_mse.py -------------------------------------------------------------------------------- /paper/experiments/02_hierarchical_time_series/02c_training_pgbm_wmse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/02_hierarchical_time_series/02c_training_pgbm_wmse.py -------------------------------------------------------------------------------- /paper/experiments/02_hierarchical_time_series/02d_training_ngboost_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/02_hierarchical_time_series/02d_training_ngboost_mse.py -------------------------------------------------------------------------------- /paper/experiments/02_hierarchical_time_series/03_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/experiments/02_hierarchical_time_series/03_evaluation.py -------------------------------------------------------------------------------- /paper/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/paper/plots.py -------------------------------------------------------------------------------- /pgbm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pgbm/sklearn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/__init__.py -------------------------------------------------------------------------------- /pgbm/sklearn/_binning.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/_binning.pyx -------------------------------------------------------------------------------- /pgbm/sklearn/_bitset.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/_bitset.pyx -------------------------------------------------------------------------------- /pgbm/sklearn/_gradient_boosting.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/_gradient_boosting.pyx -------------------------------------------------------------------------------- /pgbm/sklearn/_predictor.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/_predictor.pyx -------------------------------------------------------------------------------- /pgbm/sklearn/binning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/binning.py -------------------------------------------------------------------------------- /pgbm/sklearn/common.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/common.pyx -------------------------------------------------------------------------------- /pgbm/sklearn/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/distributions.py -------------------------------------------------------------------------------- /pgbm/sklearn/gradient_boosting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/gradient_boosting.py -------------------------------------------------------------------------------- /pgbm/sklearn/grower.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/grower.py -------------------------------------------------------------------------------- /pgbm/sklearn/histogram.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/histogram.pyx -------------------------------------------------------------------------------- /pgbm/sklearn/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/predictor.py -------------------------------------------------------------------------------- /pgbm/sklearn/splitting.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/splitting.pyx -------------------------------------------------------------------------------- /pgbm/sklearn/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_binning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_binning.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_bitset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_bitset.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_compare_lightgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_compare_lightgbm.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_estimator.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_gradient_boosting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_gradient_boosting.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_grower.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_grower.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_histogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_histogram.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_monotonic_contraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_monotonic_contraints.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_predictor.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_splitting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_splitting.py -------------------------------------------------------------------------------- /pgbm/sklearn/tests/test_warm_start.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/tests/test_warm_start.py -------------------------------------------------------------------------------- /pgbm/sklearn/utils.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/sklearn/utils.pyx -------------------------------------------------------------------------------- /pgbm/torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/torch/__init__.py -------------------------------------------------------------------------------- /pgbm/torch/pgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/torch/pgbm.py -------------------------------------------------------------------------------- /pgbm/torch/pgbm_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/torch/pgbm_dist.py -------------------------------------------------------------------------------- /pgbm/torch/splitgain_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/torch/splitgain_cpu.cpp -------------------------------------------------------------------------------- /pgbm/torch/splitgain_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/torch/splitgain_cuda.cpp -------------------------------------------------------------------------------- /pgbm/torch/splitgain_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/torch/splitgain_kernel.cu -------------------------------------------------------------------------------- /pgbm/torch/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pgbm/torch/tests/test_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pgbm/torch/tests/test_estimator.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elephaint/pgbm/HEAD/setup.py --------------------------------------------------------------------------------