ard" class="code-copy-button"><i class="bi"></i></button></pre></div>
</div>
<p>Once the training is finished, you can get the policy of the agent (as <span class="math inline">\(\pi(\uparrow|n)\)</span>) at any of the episodes in which the memory was saved by running:</p>
<div class="cell">
<div class="sourceCode cell-code" id="cb11"><pre class="sourceCode python code-with-copy"><code class="sourceCode python"><span id="cb11-1"><a href="#cb11-1" aria-hidden="true" tabindex="-1"></a><span class="im">from</s