Merge pull request optuna#830 from Crissman/integrations

Add integration modules list to readme
laranea · Jan 9, 2020 · 7aaccbe · 7aaccbe
2 parents 05d33b9 + 61e6f88
commit 7aaccbe
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -75,6 +75,21 @@ study.optimize(objective, n_trials=100)  # Invoke optimization of the objective
 ```
 
 
+## Integrations
+
+[Integrations modules](https://optuna.readthedocs.io/en/stable/tutorial/pruning.html), which allow pruning, or early stopping, of unpromising trials are available for the following libraries:
+
+* XGBoost
+* LightGBM
+* Chainer 
+* Keras
+* TensorFlow
+* tf.keras
+* MXNet
+* PyTorch Ignite
+* PyTorch Lightning
+* FastAI
+
 ## Installation
 
 Optuna is available at [the Python Package Index](https://pypi.org/project/optuna/) and on [Anaconda Cloud](https://anaconda.org/conda-forge/optuna).