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

[RLlib] Small SlateQ example fix. (#26948)

Browse source
This commit is contained in:
Artur Niederfahrenhorst 2022-07-25 15:12:42 +02:00 • committed by GitHub
parent 41c9ef709a
commit b1594260ba
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/examples/recommender_system_with_recsim_and_slateq.py
View file

@ -181,7 +181,7 @@ def main():
if args.run == "DQN":
trainer = dqn.DQN(config=config)
else:
trainer = slateq.SlateQTrainer(config=config)
trainer = slateq.SlateQ(config=config)
for i in range(10):
result = trainer.train()
print(pretty_print(result))