Skip to content
Snippets Groups Projects
Commit b93db13e authored by Bas Nijholt's avatar Bas Nijholt
Browse files

0D: return loss_improvements in choose_points

parent b0276c52
No related branches found
No related tags found
1 merge request!7implement 2D learner
......@@ -131,9 +131,7 @@ class AverageLearner(BaseLearner):
def choose_points(self, n, add_data=True):
points = list(range(self.n_requested, self.n_requested + n))
# XXX: We should return the actual loss improvements
loss_improvements = [None] * n
loss_improvements = [self.loss()] * n
if add_data:
self.add_data(points, itertools.repeat(None))
return points, loss_improvements
......
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