cprofile_get_rank_info.py

import numpy as np
from datetime import datetime
import optuna
import cProfile
import time
from optuna.visualization._rank import _get_rank_info

def sphere(trial: optuna.Trial) -> float:
    X = np.asarray([trial.suggest_float(f"x{i}", -5, 5) for i in range(10)])
    return np.sum(X ** 2)

study = optuna.create_study(sampler=optuna.samplers.RandomSampler(seed=1))
study.optimize(sphere, n_trials=10000)

profiler = cProfile.Profile()
profiler.runcall(_get_rank_info, study, None, None, "Objective Value")
profiler.dump_stats("./get_rank_info.stats")

$ git checkout master
$ git rev-parse HEAD
9b5c8635f3b1ef23c80a12cdafda4055490b72db
$ python cprofile_get_rank_info.py
$ gprof2dot -f pstats get_rank_info.stat | dot -Tpng -o get_rank_info.png