bug fix for model_plots when specifying which_data_ycols

This commit is contained in:
Zhenwen Dai 2015-09-21 16:38:06 +01:00
parent f3975e98c3
commit 4fa8005800

View file

@ -63,7 +63,7 @@ def plot_data(model, which_data_rows='all',
for d in which_data_ycols:
plots['dataplot'].append(ax.plot(X[which_data_rows, free_dims], Y[which_data_rows, d], data_symbol, mew=mew))
if X_variance is not None:
plots['xerrorbar'] = ax.errorbar(X[which_data_rows, free_dims].flatten(), Y[which_data_rows, which_data_ycols].flatten(),
plots['xerrorbar'] = ax.errorbar(X[which_data_rows, free_dims].flatten(), Y[which_data_rows, d].flatten(),
xerr=2 * np.sqrt(X_variance[which_data_rows, free_dims].flatten()),
ecolor='k', fmt='none', elinewidth=.5, alpha=.5)