Convergence Plot Callback for skopt

convergence_plot_callback.py

from skopt import gp_minimize
from skopt.plots import plot_convergence

from IPython.display import clear_output

class ConvergencePlotCallback(object):
    def __init__(self, figsize=(12,8)):
        self.fig = plt.figure(figsize=figsize)

    def __call__(self, res):
        clear_output(wait=True)
        plot_convergence(res)
        plt.show()
        
plot_callback = ConvergencePlotCallback(figsize=(12,8))
gp_results = gp_minimize(objective, space, n_calls=10, random_state=2019, callback=[plot_callback])