Skip to content
Snippets Groups Projects
Commit 4ae5d165 authored by Jorn Hoofwijk's avatar Jorn Hoofwijk Committed by Bas Nijholt
Browse files

partially solve the issue

parent 9388b52e
No related branches found
No related tags found
No related merge requests found
......@@ -153,6 +153,12 @@ class Learner1D(BaseLearner):
self.update_interpolated_loss_in_interval(x, x_right)
self.losses.pop((x_left, x_right), None)
self.losses_combined.pop((x_left, x_right), None)
a, b = self.find_neighbors(x, self.neighbors_combined)
self.losses_combined.pop((a, b), None)
if x_left is None and a is not None:
self.losses_combined[x, b] = float('inf')
if x_right is None and b is not None:
self.losses_combined[x, b] = float('inf')
else:
losses_combined = self.losses_combined
x_left, x_right = self.find_neighbors(x, self.neighbors)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment