Построение прогресса минимизации fmin () - PullRequest
Новая
       10

Построение прогресса минимизации fmin ()

0 голосов
/ 11 марта 2019

Я использую scipy's fmin для минимизации функции. Есть ли способ напечатать различные этапы минимизации поверх одного и того же графика, чтобы я мог видеть прогресс? Это функция, которую я пытаюсь минимизировать: 

import numpy as np 
import matplotlib.pylab as plt

def wave_func(func_params):
    """This function calculates the difference between a sinewave (sin(x)) and raw_data (different sin wave)
    This is the function that will be minimized by modulating a, b, k, and h parameters in order to minimize
    the difference between curves."""

    a = 1
    b = 1
    k = 0
    h = 0

    y_wave = a * np.sin((x_vals-h)/b) + k
    error = np.sum((y_wave - raw_data) * (y_wave - raw_data))
    plt.plot(y_wave)
    plt.show()
    time.sleep(1)
    return error

Тем не менее, каждый раз это строит новый сюжет, и за ним невозможно следить.

...