def vb_plot_variables(fit, var_names): samples, means, names = vb_extract(fit) if type(var_names) == str: var_names = [var_names] elif type(var_names) != list: raise Exception("Invalid argument type for var_names") to_plot = [] for var in var_names: for i in xrange(len(fit["sampler_param_names"])): if var in fit["sampler_param_names"][i]: to_plot.append(fit["sampler_param_names"][i]) for var in to_plot: plots.kdeplot_op(plt, samples[var]) plt.legend(to_plot) plt.show()
def plot_kde(samples): plots.kdeplot_op(plt, samples)