├── .gitignore ├── LICENSE ├── README.md ├── dev_requirements.txt ├── examples ├── __init__.py ├── example_classification.py └── example_regression.py ├── pyproject.toml ├── requirements.txt ├── tests ├── __init__.py ├── test_booster_objectives.py ├── test_example_classification.py ├── test_example_regression.py └── utils │ ├── __init__.py │ ├── derivative_test_params.py │ └── official_catboost_objectives.py └── treeboost_autograd ├── __init__.py ├── booster_objectives.py └── pytorch_objective.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/README.md -------------------------------------------------------------------------------- /dev_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/dev_requirements.txt -------------------------------------------------------------------------------- /examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/example_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/examples/example_classification.py -------------------------------------------------------------------------------- /examples/example_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/examples/example_regression.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | torch 3 | -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_booster_objectives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/tests/test_booster_objectives.py -------------------------------------------------------------------------------- /tests/test_example_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/tests/test_example_classification.py -------------------------------------------------------------------------------- /tests/test_example_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/tests/test_example_regression.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/utils/derivative_test_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/tests/utils/derivative_test_params.py -------------------------------------------------------------------------------- /tests/utils/official_catboost_objectives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/tests/utils/official_catboost_objectives.py -------------------------------------------------------------------------------- /treeboost_autograd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/treeboost_autograd/__init__.py -------------------------------------------------------------------------------- /treeboost_autograd/booster_objectives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/treeboost_autograd/booster_objectives.py -------------------------------------------------------------------------------- /treeboost_autograd/pytorch_objective.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TomerRonen34/treeboost_autograd/HEAD/treeboost_autograd/pytorch_objective.py --------------------------------------------------------------------------------