Increment in the i in the Dataset class

[davidsuculum]

In the next line in the Dataset class:

y = np.squeeze(data_preprocessing(np.array(self.dataset[self.keys[i+3]])[np.newaxis,:,:]))

I wonder if if instead of i+3 it wouldn't be i+4. Because, if I understand it correctly, RainNet takes the last 4 timesteps and predicts the next one (the 5th). So, with i+3 we are still in one of the data points in the x array, that runs in range(4), so from 0 to 3. Am I right or am I misunderstanding something? Thanks, @PradyumnaGupta

[PradyumnaGupta]

Nice catch @davidsuculum, yes it should be i+4. Must have made this mistake when I was cleaning the code before pushing it. I will push an update for this soon.

[kaelyavel]

Nice catch @davidsuculum, yes it should be i+4. Must have made this mistake when I was cleaning the code before pushing it. I will push an update for this soon.

@PradyumnaGupta This isn't update yet.
Is i+3 the correct version ?