Merge main

optuna · Feb 8, 2024 · a49bbcc · a49bbcc
2 parents 3e64bb3 + ab25005
commit a49bbcc
Show file tree

Hide file tree

Showing 4 changed files with 359 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -28,6 +28,7 @@ Optuna-Integration API reference is [here](https://optuna-integration.readthedoc
 * [Keras](https://optuna-integration.readthedocs.io/en/stable/reference/index.html#keras)  ([example](https://github.com/optuna/optuna-examples/tree/main/keras))
 * [MXNet](https://optuna-integration.readthedocs.io/en/stable/reference/index.html#mxnet)  ([example](https://github.com/optuna/optuna-examples/tree/main/mxnet))
 * [pycma](https://optuna-integration.readthedocs.io/en/stable/reference/index.html#pycma)
+* [scikit-optimize](https://optuna-integration.readthedocs.io/en/stable/reference/index.html#skopt)
 * [SHAP](https://optuna-integration.readthedocs.io/en/stable/reference/index.html#shap)
 * [sklearn](https://optuna-integration.readthedocs.io/en/stable/reference/index.html#sklearn)  ([example](https://github.com/optuna/optuna-examples/tree/main/sklearn/sklearn_optuna_search_cv_simple.py))
 * [skorch](https://optuna-integration.readthedocs.io/en/stable/reference/index.html#skorch)  ([example](https://github.com/optuna/optuna-examples/tree/main/pytorch/skorch_simple.py))
@@ -55,8 +56,8 @@ Also, we also provide Optuna docker images on [DockerHub](https://hub.docker.com
 
 ## Communication
 
-- [GitHub Discussions] for questions.
-- [GitHub Issues] for bug reports and feature requests.
+* [GitHub Discussions] for questions.
+* [GitHub Issues] for bug reports and feature requests.
 
 [GitHub Discussions]: https://github.com/optuna/optuna-integration/discussions
 [GitHub issues]: https://github.com/optuna/optuna-integration/issues

diff --git a/docs/source/reference/index.rst b/docs/source/reference/index.rst
@@ -111,14 +111,22 @@ MXNet
 
 pycma
 -----
-
 .. autosummary::
    :toctree: generated/
    :nosignatures:
 
    optuna_integration.CmaEsSampler
    optuna_integration.PyCmaSampler
 
+scikit-optimize
+---------------
+
+.. autosummary::
+   :toctree: generated/
+   :nosignatures:
+
+   optuna_integration.SkoptSampler
+
 SHAP
 ----
 
@@ -162,7 +170,7 @@ TensorFlow
    :toctree: generated/
    :nosignatures:
 
-   optuna.integration.TFKerasPruningCallback
+   optuna_integration.TFKerasPruningCallback
 
 XGBoost
 -------

diff --git a/optuna_integration/__init__.py b/optuna_integration/__init__.py
@@ -20,6 +20,7 @@
     "mxnet": ["MXNetPruningCallback"],
     "shap": ["ShapleyImportanceEvaluator"],
     "sklearn": ["OptunaSearchCV"],
+    "skopt": ["SkoptSampler"],
     "skorch": ["SkorchPruningCallback"],
     "tensorboard": ["TensorBoardCallback"],
     "tensorflow": ["TensorFlowPruningHook"],
@@ -46,6 +47,7 @@
     from optuna_integration.mxnet import MXNetPruningCallback
     from optuna_integration.shap import ShapleyImportanceEvaluator
     from optuna_integration.sklearn import OptunaSearchCV
+    from optuna_integration.skopt import SkoptSampler
     from optuna_integration.skorch import SkorchPruningCallback
     from optuna_integration.tensorboard import TensorBoardCallback
     from optuna_integration.tensorflow import TensorFlowPruningHook
@@ -116,6 +118,7 @@ def _get_module(self, module_name: str) -> ModuleType:
     "CmaEsSampler",
     "PyCmaSampler",
     "ShapleyImportanceEvaluator",
+    "SkoptSampler",
     "SkorchPruningCallback",
     "TensorBoardCallback",
     "TensorFlowPruningHook",