Bug fix in the contextual bandit's linear_regression.py model. (#8815)

2026-08-12 12:20:11 +08:00 · 2020-06-06 22:47:42 +02:00
parent be26a7b1b0
commit ad695a818b
1 changed files with 1 additions and 1 deletions
@@ -34,7 +34,7 @@ class OnlineLinearRegression(nn.Module):
    def partial_fit(self, x, y):
        # TODO: Handle batch of data rather than individual points
        self._check_inputs(x, y)
-        x = x.squeeze()
+        x = x.squeeze(0)
        y = y.item()
        self.time += 1
        self.delta_f += y * x