n><span class="n">plot_model</span><span class="p">(</span><span class="n">model</span><span class="o">.</span><span class="n">model</span><span class="p">,</span> <span class="n">show_shapes</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
</pre></div>
</div>
<img alt="../_images/4_train_example_8_0.png" src="../_images/4_train_example_8_0.png" />
<p>Create the dataset</p>
<div class="highlight-ipython3 notranslate"><div class="highlight"><pre><span></span><span class="n">