Bug when plotting lamb results

[adam-hartshorne]

I believe there is a bug in the code that plots the lamb results somewhere.

Using the code provided in the text notebooks, I have used a simple hand-crafted dataset, where I have added a very small amount of gaussian noise to the sequence shown in the first image (green images are the ones to be predicted).

With normalisation turned off, the resultant lamb diagram looks like the x-y axis is flipped.

With normalisation turned on, again it looks flipped but also the scaling is off,