Resolve "(Learner1D) add possibility to use the direct neighbors in the loss"

Closes #119 (closed)

Currently works for one R^1 \to R^1

Still have to make it work for R^1 \to R^N

Also performance is actually quite good. As in: the learner slows down about 1.5 times. Going (on my laptop) from 3 seconds per 1000 points to 4.5 seconds per 1000 points.

Which, I believe, will be more than compensated by the fact that the chosen points are generally better

adaptive/learner/learner1D.py

@@ -3,12 +3,14 @@ from copy import deepcopy
 import heapq
 import itertools
 import math
+from collections import Iterable
 
 import numpy as np
 import sortedcontainers
 
 from .base_learner import BaseLearner
 from .learnerND import volume
+from .triangulation import simplex_volume_in_embedding
 from ..notebook_integration import ensure_holoviews
 from ..utils import cache_latest
 
@@ -58,9 +60,14 @@ def default_loss(interval, scale, function_values):
 
 
 def _loss_of_multi_interval(xs, ys):
-    pts = list(zip(xs, ys))
-    N = len(pts) - 2
-    return sum(volume(pts[i:i+3]) for i in range(N)) / N
+    N = len(xs) - 2
+    if isinstance(ys[0], Iterable):
+        pts = [(x, *y) for x, y in zip(xs, ys)]
+        # print(pts)
+        return sum(simplex_volume_in_embedding(pts[i:i+3]) for i in range(N)) / N
+    else:
+        pts = [(x, y) for x, y in zip(xs, ys)]
+        return sum(volume(pts[i:i+3]) for i in range(N)) / N
 
 
 def triangle_loss(interval, neighbours, scale, function_values):
@@ -84,9 +91,9 @@ def get_curvature_loss(area_factor=1, euclid_factor=0.02, horizontal_factor=0.02
     def curvature_loss(interval, neighbours, scale, function_values):
         triangle_loss_ = triangle_loss(interval, neighbours, scale, function_values)
         default_loss_ = default_loss(interval, scale, function_values)
-        dx = interval[1] - interval[0] / scale[0]
+        dx = (interval[1] - interval[0]) / scale[0]
         return (area_factor * (triangle_loss_**0.5)
-                + euclid_factor * old_loss
+                + euclid_factor * default_loss_
                 + horizontal_factor * dx)
     return curvature_loss
 
@@ -156,7 +163,7 @@ class Learner1D(BaseLearner):
         self._loss_depends_on_neighbours = loss_depends_on_neighbours
 
         if loss_depends_on_neighbours:
-            self.loss_per_interval = loss_per_interval or curvature_loss
+            self.loss_per_interval = loss_per_interval or get_curvature_loss()
         else:
             self.loss_per_interval = loss_per_interval or default_loss
 
@@ -305,7 +312,7 @@ class Learner1D(BaseLearner):
     @staticmethod
     def _find_neighbors(x, neighbors):
         if x in neighbors:
-            return neighbors
+            return neighbors[x]
         if x is None:
             return None, None
         pos = neighbors.bisect_left(x)