if model == "rnn": X = H[:, 0, :-1:, :] Y = H[:, -n_outputs:, :] Is there a problem with the line of code X? Dear author, could you please share the code for the other comparison models as well? I would be extremely grateful.