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)