├── .dockerignore ├── .github └── workflows │ ├── build_and_test.yml │ └── release.yml ├── .gitignore ├── .gitmodules ├── .readthedocs.yml ├── .tool-versions ├── CHANGELOG.rst ├── Dockerfile ├── LICENSE.txt ├── Makefile ├── README.rst ├── docs ├── Makefile ├── _static │ └── .gitkeep ├── _templates │ └── .gitkeep ├── conf.py ├── ensemble │ ├── forest_boosted_regressor.rst │ ├── forest_causal_regressor.rst │ ├── forest_classifier.rst │ ├── forest_instrumental_regressor.rst │ ├── forest_local_linear_regressor.rst │ ├── forest_quantile_regressor.rst │ ├── forest_regressor.rst │ └── forest_survival.rst ├── index.rst ├── make.bat ├── requirements.txt └── tree │ ├── tree_causal_regressor.rst │ ├── tree_classifier.rst │ ├── tree_instrumental_regressor.rst │ ├── tree_interface.rst │ ├── tree_local_linear_regressor.rst │ ├── tree_quantile_regressor.rst │ ├── tree_regressor.rst │ └── tree_survival.rst ├── poetry.lock ├── pyproject.toml ├── skgrf ├── __init__.py ├── _version.py ├── base.py ├── ensemble │ ├── __init__.py │ ├── base.py │ ├── boosted_regressor.py │ ├── causal_regressor.py │ ├── classifier.py │ ├── instrumental_regressor.py │ ├── local_linear_regressor.py │ ├── quantile_regressor.py │ ├── regressor.py │ └── survival.py ├── grf.pyx ├── grf_.pxd ├── tree │ ├── __init__.py │ ├── _tree.py │ ├── base.py │ ├── causal_regressor.py │ ├── classifier.py │ ├── instrumental_regressor.py │ ├── local_linear_regressor.py │ ├── quantile_regressor.py │ ├── regressor.py │ └── survival.py └── utils │ ├── __init__.py │ ├── shap.py │ └── validation.py └── tests ├── __init__.py ├── _fixtures ├── __init__.py ├── causalml │ ├── LICENSE.txt │ ├── README.md │ ├── __init__.py │ └── regression.py └── sksurv │ ├── LICENSE.txt │ ├── README.md │ └── veteran.arff ├── conftest.py ├── ensemble ├── __init__.py ├── test_boosted_regressor.py ├── test_causal_regressor.py ├── test_classifier.py ├── test_instrumental_regressor.py ├── test_local_linear_regressor.py ├── test_quantile_regressor.py ├── test_regressor.py └── test_survival.py ├── test_tools.py └── tree ├── __init__.py ├── test_causal_regressor.py ├── test_classifier.py ├── test_instrumental_regressor.py ├── test_local_linear_regressor.py ├── test_quantile_regressor.py ├── test_regressor.py └── test_survival.py /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/.dockerignore -------------------------------------------------------------------------------- /.github/workflows/build_and_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/.github/workflows/build_and_test.yml -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/.gitmodules -------------------------------------------------------------------------------- /.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/.readthedocs.yml -------------------------------------------------------------------------------- /.tool-versions: -------------------------------------------------------------------------------- 1 | python 3.8.5 2 | poetry 1.1.12 3 | -------------------------------------------------------------------------------- /CHANGELOG.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/CHANGELOG.rst -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/Makefile -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/README.rst -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/_templates/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/ensemble/forest_boosted_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/ensemble/forest_boosted_regressor.rst -------------------------------------------------------------------------------- /docs/ensemble/forest_causal_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/ensemble/forest_causal_regressor.rst -------------------------------------------------------------------------------- /docs/ensemble/forest_classifier.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/ensemble/forest_classifier.rst -------------------------------------------------------------------------------- /docs/ensemble/forest_instrumental_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/ensemble/forest_instrumental_regressor.rst -------------------------------------------------------------------------------- /docs/ensemble/forest_local_linear_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/ensemble/forest_local_linear_regressor.rst -------------------------------------------------------------------------------- /docs/ensemble/forest_quantile_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/ensemble/forest_quantile_regressor.rst -------------------------------------------------------------------------------- /docs/ensemble/forest_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/ensemble/forest_regressor.rst -------------------------------------------------------------------------------- /docs/ensemble/forest_survival.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/ensemble/forest_survival.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/tree/tree_causal_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/tree/tree_causal_regressor.rst -------------------------------------------------------------------------------- /docs/tree/tree_classifier.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/tree/tree_classifier.rst -------------------------------------------------------------------------------- /docs/tree/tree_instrumental_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/tree/tree_instrumental_regressor.rst -------------------------------------------------------------------------------- /docs/tree/tree_interface.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/tree/tree_interface.rst -------------------------------------------------------------------------------- /docs/tree/tree_local_linear_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/tree/tree_local_linear_regressor.rst -------------------------------------------------------------------------------- /docs/tree/tree_quantile_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/tree/tree_quantile_regressor.rst -------------------------------------------------------------------------------- /docs/tree/tree_regressor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/tree/tree_regressor.rst -------------------------------------------------------------------------------- /docs/tree/tree_survival.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/docs/tree/tree_survival.rst -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/pyproject.toml -------------------------------------------------------------------------------- /skgrf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/__init__.py -------------------------------------------------------------------------------- /skgrf/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.3.0" 2 | -------------------------------------------------------------------------------- /skgrf/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/base.py -------------------------------------------------------------------------------- /skgrf/ensemble/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/__init__.py -------------------------------------------------------------------------------- /skgrf/ensemble/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/base.py -------------------------------------------------------------------------------- /skgrf/ensemble/boosted_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/boosted_regressor.py -------------------------------------------------------------------------------- /skgrf/ensemble/causal_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/causal_regressor.py -------------------------------------------------------------------------------- /skgrf/ensemble/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/classifier.py -------------------------------------------------------------------------------- /skgrf/ensemble/instrumental_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/instrumental_regressor.py -------------------------------------------------------------------------------- /skgrf/ensemble/local_linear_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/local_linear_regressor.py -------------------------------------------------------------------------------- /skgrf/ensemble/quantile_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/quantile_regressor.py -------------------------------------------------------------------------------- /skgrf/ensemble/regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/regressor.py -------------------------------------------------------------------------------- /skgrf/ensemble/survival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/ensemble/survival.py -------------------------------------------------------------------------------- /skgrf/grf.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/grf.pyx -------------------------------------------------------------------------------- /skgrf/grf_.pxd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/grf_.pxd -------------------------------------------------------------------------------- /skgrf/tree/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/__init__.py -------------------------------------------------------------------------------- /skgrf/tree/_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/_tree.py -------------------------------------------------------------------------------- /skgrf/tree/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/base.py -------------------------------------------------------------------------------- /skgrf/tree/causal_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/causal_regressor.py -------------------------------------------------------------------------------- /skgrf/tree/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/classifier.py -------------------------------------------------------------------------------- /skgrf/tree/instrumental_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/instrumental_regressor.py -------------------------------------------------------------------------------- /skgrf/tree/local_linear_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/local_linear_regressor.py -------------------------------------------------------------------------------- /skgrf/tree/quantile_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/quantile_regressor.py -------------------------------------------------------------------------------- /skgrf/tree/regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/regressor.py -------------------------------------------------------------------------------- /skgrf/tree/survival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/tree/survival.py -------------------------------------------------------------------------------- /skgrf/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /skgrf/utils/shap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/utils/shap.py -------------------------------------------------------------------------------- /skgrf/utils/validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/skgrf/utils/validation.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/_fixtures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/_fixtures/causalml/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/_fixtures/causalml/LICENSE.txt -------------------------------------------------------------------------------- /tests/_fixtures/causalml/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/_fixtures/causalml/README.md -------------------------------------------------------------------------------- /tests/_fixtures/causalml/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/_fixtures/causalml/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/_fixtures/causalml/regression.py -------------------------------------------------------------------------------- /tests/_fixtures/sksurv/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/_fixtures/sksurv/LICENSE.txt -------------------------------------------------------------------------------- /tests/_fixtures/sksurv/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/_fixtures/sksurv/README.md -------------------------------------------------------------------------------- /tests/_fixtures/sksurv/veteran.arff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/_fixtures/sksurv/veteran.arff -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/ensemble/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ensemble/test_boosted_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/ensemble/test_boosted_regressor.py -------------------------------------------------------------------------------- /tests/ensemble/test_causal_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/ensemble/test_causal_regressor.py -------------------------------------------------------------------------------- /tests/ensemble/test_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/ensemble/test_classifier.py -------------------------------------------------------------------------------- /tests/ensemble/test_instrumental_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/ensemble/test_instrumental_regressor.py -------------------------------------------------------------------------------- /tests/ensemble/test_local_linear_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/ensemble/test_local_linear_regressor.py -------------------------------------------------------------------------------- /tests/ensemble/test_quantile_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/ensemble/test_quantile_regressor.py -------------------------------------------------------------------------------- /tests/ensemble/test_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/ensemble/test_regressor.py -------------------------------------------------------------------------------- /tests/ensemble/test_survival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/ensemble/test_survival.py -------------------------------------------------------------------------------- /tests/test_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/test_tools.py -------------------------------------------------------------------------------- /tests/tree/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/tree/test_causal_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/tree/test_causal_regressor.py -------------------------------------------------------------------------------- /tests/tree/test_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/tree/test_classifier.py -------------------------------------------------------------------------------- /tests/tree/test_instrumental_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/tree/test_instrumental_regressor.py -------------------------------------------------------------------------------- /tests/tree/test_local_linear_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/tree/test_local_linear_regressor.py -------------------------------------------------------------------------------- /tests/tree/test_quantile_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/tree/test_quantile_regressor.py -------------------------------------------------------------------------------- /tests/tree/test_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/tree/test_regressor.py -------------------------------------------------------------------------------- /tests/tree/test_survival.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crflynn/skgrf/HEAD/tests/tree/test_survival.py --------------------------------------------------------------------------------