mlflow-xgboost-proba 0.1.0

MLflow XGBoost flavour with probabilities

mlflow-xgboost-proba

MLflow XGBoost flavour with probabilities

This package implements mlflow_xgboost_proba MLflow flavour, which allows to run predict_proba method of xgboost models during inference with MLflow mlflow models serve CLI command.

Implementation is based on mlflow.xgboost module, which is copied and modified to have the wrapper with predict_proba method and predict method calling predict_proba by default.

The API of the module is identical to mlflow.xgboost, only without support of autologging.

• Github repository: https://github.com/sergray/mlflow-xgboost-proba/

Getting started with your project

First, create a repository on GitHub with the same name as this project, and then run the following commands:

git init -b main
git add .
git commit -m "init commit"
git remote add origin [e-mail]:sergray/mlflow-xgboost-proba.git
git push -u origin main

Finally, install the environment and the pre-commit hooks with

make install

You are now ready to start development on your project! The CI/CD pipeline will be triggered when you open a pull request, merge to main, or when you create a new release.

Releasing a new version

• Create an API Token on Pypi.
• Add the API Token to your projects secrets with the name PYPI_TOKEN by visiting this page.
• Create a new release on Github.
• Create a new tag in the form *.*.*.

---

Repository initiated with fpgmaas/cookiecutter-poetry.