<span class="s1">'tf'</span><span class="p">)</span>

    <span class="c1"># Return the keras model</span>
    <span class="k">return</span> <span class="n">embedding_network</span>

<span class="n">my_model</span><span class="o">.</span><span class="n">on_save_keras_model</span> <span class="o">=</span> <span class="n">my_keras_model_saver</span>
</pre></div>
</div>
</section>
<section id="train-the-model">
<h3 id="train-the-model">Train the model<a class="headerlink" href="#train-the-model" title="Permali