hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Sven Mika 2020-06-06 22:47:42 +02:00 • committed by GitHub
parent be26a7b1b0
commit ad695a818b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
rllib/contrib/bandits/models/linear_regression.py
View file

@ -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