[備忘録] Optunaの探索結果をUMAPで可視化する - Qiita

ハイパパラメータが3つ以上存在する場合，探索結果の可視化はなかなか難しいです． 今回はUMAPを利用して探索結果の可視化を行います． import matplotlib.pyplot as plt import numpy as np import optuna import umap DIM = 6 SEED = 42 N_SAMPLES = 1000 N_GRID = 50 LB, UB = -5, 5 def func(X: np.ndarray) -> np.ndarray: return np.sum(X ** 2, axis=-1) def objective(trial: optuna.Trial) -> float: X = np.asarray([trial.suggest_float(f"x{d}", LB, UB) for d in range(DIM)]) retu