WIP: add support for neighbours in loss computation in LearnerND
Closes #120
TODO add support to output in R^N
TODO rewrite the code to be more readable, I will do this next week
As you can see in the plot, it is getting hard to distinguish the LearnerND from the Learner2D :D