Refactor prediction

Created by: lu-zero

As @luctrudeau suggested.

We'll spare the edges generation by using split_at-like functions over the slices (so we can borrow them multiple times when needed) later.