Gurobi Machine Learning

A Python package to help use trained regression models in mathematical optimization models. The package supports a variety of regression models (linear, logistic, neural networks, decision trees,…) trained by different machine learning frameworks (scikit-learn, LightGBM, XGBoost, Keras and PyTorch).

Get Started

Contents

• User Guide
  • Introduction
  • Usage Example
  • Supported Regression models
  • Mixed Integer Formulations
  • Further Reading
  • Bibliography
• Examples
• API
  • Main Function
  • Scikit-learn API
  • Keras API
  • PyTorch API
  • XGBoost API
  • LightGBM API
• Contact Us
• License