1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
"""
plot_rank
=========
.. autofunction:: optuna.visualization.plot_rank
The following code snippet shows how to plot the parameter relationship as a rank plot.
"""
import optuna
from plotly.io import show
def objective(trial):
x = trial.suggest_float("x", -100, 100)
y = trial.suggest_categorical("y", [-1, 0, 1])
c0 = 400 - (x + y) ** 2
trial.set_user_attr("constraint", [c0])
return x**2 + y
def constraints(trial):
return trial.user_attrs["constraint"]
sampler = optuna.samplers.TPESampler(seed=10, constraints_func=constraints)
study = optuna.create_study(sampler=sampler)
study.optimize(objective, n_trials=30)
fig = optuna.visualization.plot_rank(study, params=["x", "y"])
show(fig)
|
