TrackSolve#

class TrackSolve(ml, tmin=None, tmax=None, update_iter=None)[source]#

Track and/or visualize optimization progress for Pastas models.

Parameters

ml (pastas.model.Model) – pastas Model to track
tmin (str or pandas.Timestamp, optional) – start time for simulation, by default None which defaults to first index in ml.oseries.series
tmax (str or pandas.Timestamp, optional) – end time for simulation, by default None which defaults to last index in ml.oseries.series
update_iter (int, optional) – if visualizing optimization progress, update plot every update_iter iterations, by default nparam

Notes

Interactive plotting of optimization progress requires a matplotlib backend that supports interactive plotting, e.g. mpl.use(“TkAgg”) and mpl.interactive( True). Some possible speedups on the matplotlib side include: - mpl.style.use(“fast”) - mpl.rcParams[‘path.simplify_threshold’] = 1.0

Examples

Set matplotlib backend and interactive mode (put this at the top of your script):

import matplotlib as mpl
mpl.use("TkAgg")
import matplotlib.pyplot as plt
plt.ion()

Create a TrackSolve object for your model:

track = TrackSolve(ml)

Solve model and store intermediate optimization results:

ml.solve(callback=track.track_solve)

Calculated parameters per iteration are stored in a pandas.DataFrame:

track.parameters

Other stored statistics include track.evp (explained variance percentage), track.rmse_res (root-mean-squared error of the residuals), track.rmse_noise ( root mean squared error of the noise, only if noise=True).

To interactively plot model optimization progress while solving pass track.plot_track_solve as callback function:

ml.solve(callback=track.plot_track_solve)

Access the resulting figure through track.fig.

Methods#

`__init__`
`initialize_figure`	Initialize figure for plotting optimization progress.
`plot_track_solve`	Method to plot model simulation while model is being solved.
`plot_track_solve_history`	Plot optimization history.
`track_solve`	Append parameters to self.parameters DataFrame and update itercount, rmse values and evp.