def get_which_data_ycols(model, which_data_ycols):
if which_data_ycols == '<STR_LIT:all>' or which_data_ycols is None:<EOL><INDENT>return np.arange(model.output_dim)<EOL><DEDENT>return which_data_ycols<EOL>
Helper to get the data columns to plot.