How to extract neural network of reinforcement learning agent?
6 views (last 30 days)
Show older comments
Yihao Wan
on 4 Mar 2021
Edited: Emmanouil Tzorakoleftherakis
on 11 Mar 2021
Is there a way to get the neural network of trained reinforcement learning agent?
0 Comments
Accepted Answer
Emmanouil Tzorakoleftherakis
on 5 Mar 2021
For critics:
criticNet = getModel(getCritic(trained_agent))
For actors
actorNet = getModel(getActor(trained_agent))
Note that you can use this regardless of whether the agent is trained or not.
Hope this helps
5 Comments
Emmanouil Tzorakoleftherakis
on 5 Mar 2021
Edited: Emmanouil Tzorakoleftherakis
on 11 Mar 2021
What I mentioned above should work then. Use the DL blocks or ML Function block to bring the trained network into Simulink, and the plain C codegen feature I mentioned in my other answer to deploy. Plain C code generation shipped in R2021a.
More Answers (0)
See Also
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!