teddy
teddy

Reputation: 2208

How to use tensorflow.Estimator for reinforcement learning

The tensorflow.Estimator seems to gear toward supervised learning but seems to difficult to adopt for other task even if only small changes are needed for the model or training. For example, in reinforcement learning, I would need to feed a reward value which is not part of the features.

Upvotes: 5

Views: 680

Answers (1)

iga
iga

Reputation: 3633

Estimators were meant for good-old supervised learning. So, adopting them to reinforcement learning will likely feel awkward. Here is an RL library that might be useful: https://github.com/tensorflow/agents.

Upvotes: 2

Related Questions