| |
| |
| |
| <!DOCTYPE html> |
| <html> |
| <head> |
| <meta http-equiv="Content-Type" content="text/html; charset=utf-8"> |
| |
| |
| <meta http-equiv="X-UA-Compatible" content="IE=emulateIE7" /> |
| <title>Coverage for marvin_image_classification_engine/training/trainer.py: 90%</title> |
| <link rel="stylesheet" href="style.css" type="text/css"> |
| |
| <script type="text/javascript" src="jquery.min.js"></script> |
| <script type="text/javascript" src="jquery.hotkeys.js"></script> |
| <script type="text/javascript" src="jquery.isonscreen.js"></script> |
| <script type="text/javascript" src="coverage_html.js"></script> |
| <script type="text/javascript"> |
| jQuery(document).ready(coverage.pyfile_ready); |
| </script> |
| </head> |
| <body class="pyfile"> |
| |
| <div id="header"> |
| <div class="content"> |
| <h1>Coverage for <b>marvin_image_classification_engine/training/trainer.py</b> : |
| <span class="pc_cov">90%</span> |
| </h1> |
| |
| <img id="keyboard_icon" src="keybd_closed.png" alt="Show keyboard shortcuts" /> |
| |
| <h2 class="stats"> |
| 50 statements |
| <span class="run hide_run shortkey_r button_toggle_run">45 run</span> |
| <span class="mis shortkey_m button_toggle_mis">5 missing</span> |
| <span class="exc shortkey_x button_toggle_exc">0 excluded</span> |
| |
| |
| </h2> |
| </div> |
| </div> |
| |
| <div class="help_panel"> |
| <img id="panel_icon" src="keybd_open.png" alt="Hide keyboard shortcuts" /> |
| <p class="legend">Hot-keys on this page</p> |
| <div> |
| <p class="keyhelp"> |
| <span class="key">r</span> |
| <span class="key">m</span> |
| <span class="key">x</span> |
| <span class="key">p</span> toggle line displays |
| </p> |
| <p class="keyhelp"> |
| <span class="key">j</span> |
| <span class="key">k</span> next/prev highlighted chunk |
| </p> |
| <p class="keyhelp"> |
| <span class="key">0</span> (zero) top of page |
| </p> |
| <p class="keyhelp"> |
| <span class="key">1</span> (one) first highlighted chunk |
| </p> |
| </div> |
| </div> |
| |
| <div id="source"> |
| <table> |
| <tr> |
| <td class="linenos"> |
| <p id="n1" class="pln"><a href="#n1">1</a></p> |
| <p id="n2" class="pln"><a href="#n2">2</a></p> |
| <p id="n3" class="pln"><a href="#n3">3</a></p> |
| <p id="n4" class="stm run hide_run"><a href="#n4">4</a></p> |
| <p id="n5" class="pln"><a href="#n5">5</a></p> |
| <p id="n6" class="pln"><a href="#n6">6</a></p> |
| <p id="n7" class="pln"><a href="#n7">7</a></p> |
| <p id="n8" class="stm run hide_run"><a href="#n8">8</a></p> |
| <p id="n9" class="stm run hide_run"><a href="#n9">9</a></p> |
| <p id="n10" class="stm run hide_run"><a href="#n10">10</a></p> |
| <p id="n11" class="stm run hide_run"><a href="#n11">11</a></p> |
| <p id="n12" class="stm run hide_run"><a href="#n12">12</a></p> |
| <p id="n13" class="stm run hide_run"><a href="#n13">13</a></p> |
| <p id="n14" class="stm run hide_run"><a href="#n14">14</a></p> |
| <p id="n15" class="stm run hide_run"><a href="#n15">15</a></p> |
| <p id="n16" class="stm run hide_run"><a href="#n16">16</a></p> |
| <p id="n17" class="stm run hide_run"><a href="#n17">17</a></p> |
| <p id="n18" class="pln"><a href="#n18">18</a></p> |
| <p id="n19" class="stm run hide_run"><a href="#n19">19</a></p> |
| <p id="n20" class="pln"><a href="#n20">20</a></p> |
| <p id="n21" class="stm run hide_run"><a href="#n21">21</a></p> |
| <p id="n22" class="pln"><a href="#n22">22</a></p> |
| <p id="n23" class="pln"><a href="#n23">23</a></p> |
| <p id="n24" class="stm run hide_run"><a href="#n24">24</a></p> |
| <p id="n25" class="pln"><a href="#n25">25</a></p> |
| <p id="n26" class="pln"><a href="#n26">26</a></p> |
| <p id="n27" class="stm run hide_run"><a href="#n27">27</a></p> |
| <p id="n28" class="stm run hide_run"><a href="#n28">28</a></p> |
| <p id="n29" class="stm run hide_run"><a href="#n29">29</a></p> |
| <p id="n30" class="pln"><a href="#n30">30</a></p> |
| <p id="n31" class="stm run hide_run"><a href="#n31">31</a></p> |
| <p id="n32" class="stm run hide_run"><a href="#n32">32</a></p> |
| <p id="n33" class="stm run hide_run"><a href="#n33">33</a></p> |
| <p id="n34" class="stm run hide_run"><a href="#n34">34</a></p> |
| <p id="n35" class="stm run hide_run"><a href="#n35">35</a></p> |
| <p id="n36" class="pln"><a href="#n36">36</a></p> |
| <p id="n37" class="stm run hide_run"><a href="#n37">37</a></p> |
| <p id="n38" class="stm run hide_run"><a href="#n38">38</a></p> |
| <p id="n39" class="stm run hide_run"><a href="#n39">39</a></p> |
| <p id="n40" class="pln"><a href="#n40">40</a></p> |
| <p id="n41" class="stm run hide_run"><a href="#n41">41</a></p> |
| <p id="n42" class="stm run hide_run"><a href="#n42">42</a></p> |
| <p id="n43" class="stm run hide_run"><a href="#n43">43</a></p> |
| <p id="n44" class="pln"><a href="#n44">44</a></p> |
| <p id="n45" class="stm run hide_run"><a href="#n45">45</a></p> |
| <p id="n46" class="stm run hide_run"><a href="#n46">46</a></p> |
| <p id="n47" class="stm run hide_run"><a href="#n47">47</a></p> |
| <p id="n48" class="stm run hide_run"><a href="#n48">48</a></p> |
| <p id="n49" class="stm run hide_run"><a href="#n49">49</a></p> |
| <p id="n50" class="stm run hide_run"><a href="#n50">50</a></p> |
| <p id="n51" class="stm run hide_run"><a href="#n51">51</a></p> |
| <p id="n52" class="pln"><a href="#n52">52</a></p> |
| <p id="n53" class="stm run hide_run"><a href="#n53">53</a></p> |
| <p id="n54" class="stm run hide_run"><a href="#n54">54</a></p> |
| <p id="n55" class="pln"><a href="#n55">55</a></p> |
| <p id="n56" class="stm run hide_run"><a href="#n56">56</a></p> |
| <p id="n57" class="stm mis"><a href="#n57">57</a></p> |
| <p id="n58" class="stm mis"><a href="#n58">58</a></p> |
| <p id="n59" class="stm mis"><a href="#n59">59</a></p> |
| <p id="n60" class="stm mis"><a href="#n60">60</a></p> |
| <p id="n61" class="stm mis"><a href="#n61">61</a></p> |
| <p id="n62" class="pln"><a href="#n62">62</a></p> |
| <p id="n63" class="pln"><a href="#n63">63</a></p> |
| <p id="n64" class="stm run hide_run"><a href="#n64">64</a></p> |
| <p id="n65" class="stm run hide_run"><a href="#n65">65</a></p> |
| <p id="n66" class="stm run hide_run"><a href="#n66">66</a></p> |
| <p id="n67" class="pln"><a href="#n67">67</a></p> |
| <p id="n68" class="pln"><a href="#n68">68</a></p> |
| <p id="n69" class="pln"><a href="#n69">69</a></p> |
| <p id="n70" class="stm run hide_run"><a href="#n70">70</a></p> |
| <p id="n71" class="stm run hide_run"><a href="#n71">71</a></p> |
| <p id="n72" class="pln"><a href="#n72">72</a></p> |
| <p id="n73" class="stm run hide_run"><a href="#n73">73</a></p> |
| <p id="n74" class="pln"><a href="#n74">74</a></p> |
| <p id="n75" class="pln"><a href="#n75">75</a></p> |
| <p id="n76" class="pln"><a href="#n76">76</a></p> |
| <p id="n77" class="pln"><a href="#n77">77</a></p> |
| <p id="n78" class="pln"><a href="#n78">78</a></p> |
| <p id="n79" class="stm run hide_run"><a href="#n79">79</a></p> |
| |
| </td> |
| <td class="text"> |
| <p id="t1" class="pln"><span class="com">#!/usr/bin/env python</span><span class="strut"> </span></p> |
| <p id="t2" class="pln"><span class="com"># coding=utf-8</span><span class="strut"> </span></p> |
| <p id="t3" class="pln"><span class="strut"> </span></p> |
| <p id="t4" class="stm run hide_run"><span class="str">"""Trainer engine action.</span><span class="strut"> </span></p> |
| <p id="t5" class="pln"><span class="strut"> </span></p> |
| <p id="t6" class="pln"><span class="str">Use this module to add the project main code.</span><span class="strut"> </span></p> |
| <p id="t7" class="pln"><span class="str">"""</span><span class="strut"> </span></p> |
| <p id="t8" class="stm run hide_run"><span class="key">import</span> <span class="nam">os</span><span class="strut"> </span></p> |
| <p id="t9" class="stm run hide_run"><span class="key">import</span> <span class="nam">numpy</span> <span class="key">as</span> <span class="nam">np</span><span class="strut"> </span></p> |
| <p id="t10" class="stm run hide_run"><span class="key">import</span> <span class="nam">cv2</span><span class="strut"> </span></p> |
| <p id="t11" class="stm run hide_run"><span class="key">from</span> <span class="nam">keras</span><span class="op">.</span><span class="nam">layers</span> <span class="key">import</span> <span class="nam">Activation</span><span class="op">,</span> <span class="nam">Reshape</span><span class="op">,</span> <span class="nam">Dropout</span><span class="op">,</span> <span class="nam">Dense</span><span class="op">,</span> <span class="nam">Flatten</span><span class="strut"> </span></p> |
| <p id="t12" class="stm run hide_run"><span class="key">from</span> <span class="nam">keras</span><span class="op">.</span><span class="nam">layers</span> <span class="key">import</span> <span class="nam">AtrousConvolution2D</span><span class="op">,</span> <span class="nam">Conv2D</span><span class="op">,</span> <span class="nam">MaxPooling2D</span><span class="op">,</span> <span class="nam">Conv2DTranspose</span><span class="op">,</span> <span class="nam">UpSampling2D</span><span class="strut"> </span></p> |
| <p id="t13" class="stm run hide_run"><span class="key">from</span> <span class="nam">keras</span><span class="op">.</span><span class="nam">models</span> <span class="key">import</span> <span class="nam">Sequential</span><span class="strut"> </span></p> |
| <p id="t14" class="stm run hide_run"><span class="key">from</span> <span class="nam">keras</span> <span class="key">import</span> <span class="nam">callbacks</span><span class="op">,</span> <span class="nam">optimizers</span><span class="strut"> </span></p> |
| <p id="t15" class="stm run hide_run"><span class="key">from</span> <span class="op">.</span><span class="op">.</span><span class="nam">model_serializer</span> <span class="key">import</span> <span class="nam">ModelSerializer</span><span class="strut"> </span></p> |
| <p id="t16" class="stm run hide_run"><span class="key">from</span> <span class="op">.</span><span class="op">.</span><span class="nam">_compatibility</span> <span class="key">import</span> <span class="nam">six</span><span class="strut"> </span></p> |
| <p id="t17" class="stm run hide_run"><span class="key">from</span> <span class="op">.</span><span class="op">.</span><span class="nam">_logging</span> <span class="key">import</span> <span class="nam">get_logger</span><span class="strut"> </span></p> |
| <p id="t18" class="pln"><span class="strut"> </span></p> |
| <p id="t19" class="stm run hide_run"><span class="key">from</span> <span class="nam">marvin_python_toolbox</span><span class="op">.</span><span class="nam">engine_base</span> <span class="key">import</span> <span class="nam">EngineBaseTraining</span><span class="strut"> </span></p> |
| <p id="t20" class="pln"><span class="strut"> </span></p> |
| <p id="t21" class="stm run hide_run"><span class="nam">__all__</span> <span class="op">=</span> <span class="op">[</span><span class="str">'Trainer'</span><span class="op">]</span><span class="strut"> </span></p> |
| <p id="t22" class="pln"><span class="strut"> </span></p> |
| <p id="t23" class="pln"><span class="strut"> </span></p> |
| <p id="t24" class="stm run hide_run"><span class="nam">logger</span> <span class="op">=</span> <span class="nam">get_logger</span><span class="op">(</span><span class="str">'trainer'</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t25" class="pln"><span class="strut"> </span></p> |
| <p id="t26" class="pln"><span class="strut"> </span></p> |
| <p id="t27" class="stm run hide_run"><span class="key">class</span> <span class="nam">Trainer</span><span class="op">(</span><span class="nam">ModelSerializer</span><span class="op">,</span> <span class="nam">EngineBaseTraining</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p> |
| <p id="t28" class="stm run hide_run"> <span class="key">def</span> <span class="nam">__init__</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="op">**</span><span class="nam">kwargs</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p> |
| <p id="t29" class="stm run hide_run"> <span class="nam">super</span><span class="op">(</span><span class="nam">Trainer</span><span class="op">,</span> <span class="nam">self</span><span class="op">)</span><span class="op">.</span><span class="nam">__init__</span><span class="op">(</span><span class="op">**</span><span class="nam">kwargs</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t30" class="pln"><span class="strut"> </span></p> |
| <p id="t31" class="stm run hide_run"> <span class="key">def</span> <span class="nam">build_model</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">w</span><span class="op">=</span><span class="num">150</span><span class="op">,</span> <span class="nam">h</span><span class="op">=</span><span class="num">150</span><span class="op">,</span> <span class="nam">trainable</span><span class="op">=</span><span class="nam">True</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p> |
| <p id="t32" class="stm run hide_run"> <span class="nam">model</span> <span class="op">=</span> <span class="nam">Sequential</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t33" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Conv2D</span><span class="op">(</span><span class="num">32</span><span class="op">,</span> <span class="op">(</span><span class="num">3</span><span class="op">,</span> <span class="num">3</span><span class="op">)</span><span class="op">,</span> <span class="nam">input_shape</span><span class="op">=</span><span class="op">(</span><span class="nam">w</span><span class="op">,</span> <span class="nam">h</span><span class="op">,</span> <span class="num">3</span><span class="op">)</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t34" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Activation</span><span class="op">(</span><span class="str">'relu'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t35" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">MaxPooling2D</span><span class="op">(</span><span class="nam">pool_size</span><span class="op">=</span><span class="op">(</span><span class="num">2</span><span class="op">,</span> <span class="num">2</span><span class="op">)</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t36" class="pln"><span class="strut"> </span></p> |
| <p id="t37" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Conv2D</span><span class="op">(</span><span class="num">32</span><span class="op">,</span> <span class="op">(</span><span class="num">3</span><span class="op">,</span> <span class="num">3</span><span class="op">)</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t38" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Activation</span><span class="op">(</span><span class="str">'relu'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t39" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">MaxPooling2D</span><span class="op">(</span><span class="nam">pool_size</span><span class="op">=</span><span class="op">(</span><span class="num">2</span><span class="op">,</span> <span class="num">2</span><span class="op">)</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t40" class="pln"><span class="strut"> </span></p> |
| <p id="t41" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Conv2D</span><span class="op">(</span><span class="num">64</span><span class="op">,</span> <span class="op">(</span><span class="num">3</span><span class="op">,</span> <span class="num">3</span><span class="op">)</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t42" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Activation</span><span class="op">(</span><span class="str">'relu'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t43" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">MaxPooling2D</span><span class="op">(</span><span class="nam">pool_size</span><span class="op">=</span><span class="op">(</span><span class="num">2</span><span class="op">,</span> <span class="num">2</span><span class="op">)</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t44" class="pln"><span class="strut"> </span></p> |
| <p id="t45" class="stm run hide_run"> <span class="key">if</span> <span class="nam">trainable</span><span class="op">:</span><span class="strut"> </span></p> |
| <p id="t46" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Flatten</span><span class="op">(</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t47" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Dense</span><span class="op">(</span><span class="num">64</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t48" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Activation</span><span class="op">(</span><span class="str">'relu'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t49" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Dropout</span><span class="op">(</span><span class="num">0.5</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t50" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Dense</span><span class="op">(</span><span class="num">1</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t51" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">add</span><span class="op">(</span><span class="nam">Activation</span><span class="op">(</span><span class="str">'softmax'</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t52" class="pln"><span class="strut"> </span></p> |
| <p id="t53" class="stm run hide_run"> <span class="key">print</span><span class="op">(</span><span class="nam">model</span><span class="op">.</span><span class="nam">summary</span><span class="op">(</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t54" class="stm run hide_run"> <span class="key">return</span> <span class="nam">model</span><span class="strut"> </span></p> |
| <p id="t55" class="pln"><span class="strut"> </span></p> |
| <p id="t56" class="stm run hide_run"> <span class="key">def</span> <span class="nam">generate_samples</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">fnames</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p> |
| <p id="t57" class="stm mis"> <span class="key">while</span> <span class="nam">True</span><span class="op">:</span><span class="strut"> </span></p> |
| <p id="t58" class="stm mis"> <span class="key">for</span> <span class="nam">fname</span><span class="op">,</span> <span class="nam">label</span> <span class="key">in</span> <span class="nam">fnames</span><span class="op">:</span><span class="strut"> </span></p> |
| <p id="t59" class="stm mis"> <span class="nam">image</span> <span class="op">=</span> <span class="nam">cv2</span><span class="op">.</span><span class="nam">imread</span><span class="op">(</span><span class="nam">fname</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t60" class="stm mis"> <span class="nam">image</span> <span class="op">=</span> <span class="nam">image</span><span class="op">[</span><span class="nam">np</span><span class="op">.</span><span class="nam">newaxis</span><span class="op">,</span> <span class="op">:</span><span class="op">,</span> <span class="op">:</span><span class="op">,</span> <span class="op">(</span><span class="num">2</span><span class="op">,</span> <span class="num">1</span><span class="op">,</span> <span class="num">0</span><span class="op">)</span><span class="op">]</span><span class="strut"> </span></p> |
| <p id="t61" class="stm mis"> <span class="key">yield</span> <span class="op">(</span><span class="nam">image</span><span class="op">,</span> <span class="nam">np</span><span class="op">.</span><span class="nam">array</span><span class="op">(</span><span class="op">[</span><span class="nam">int</span><span class="op">(</span><span class="nam">label</span><span class="op">)</span><span class="op">]</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t62" class="pln"><span class="strut"> </span></p> |
| <p id="t63" class="pln"><span class="strut"> </span></p> |
| <p id="t64" class="stm run hide_run"> <span class="key">def</span> <span class="nam">execute</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">params</span><span class="op">,</span> <span class="op">**</span><span class="nam">kwargs</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p> |
| <p id="t65" class="stm run hide_run"> <span class="nam">model</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">build_model</span><span class="op">(</span><span class="nam">trainable</span><span class="op">=</span><span class="nam">True</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t66" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">compile</span><span class="op">(</span><span class="nam">loss</span><span class="op">=</span><span class="str">'binary_crossentropy'</span><span class="op">,</span><span class="strut"> </span></p> |
| <p id="t67" class="pln"> <span class="nam">optimizer</span><span class="op">=</span><span class="nam">optimizers</span><span class="op">.</span><span class="nam">SGD</span><span class="op">(</span><span class="nam">lr</span><span class="op">=</span><span class="nam">params</span><span class="op">[</span><span class="str">'LEARNING_RATE'</span><span class="op">]</span><span class="op">,</span> <span class="nam">momentum</span><span class="op">=</span><span class="nam">params</span><span class="op">[</span><span class="str">'MOMENTUM'</span><span class="op">]</span><span class="op">)</span><span class="op">,</span><span class="strut"> </span></p> |
| <p id="t68" class="pln"> <span class="nam">metrics</span><span class="op">=</span><span class="op">[</span><span class="str">'accuracy'</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t69" class="pln"><span class="strut"> </span></p> |
| <p id="t70" class="stm run hide_run"> <span class="nam">training_data</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">generate_samples</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">marvin_dataset</span><span class="op">[</span><span class="str">'train'</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t71" class="stm run hide_run"> <span class="nam">validation_data</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">generate_samples</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">marvin_dataset</span><span class="op">[</span><span class="str">'val'</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t72" class="pln"><span class="strut"> </span></p> |
| <p id="t73" class="stm run hide_run"> <span class="nam">model</span><span class="op">.</span><span class="nam">fit_generator</span><span class="op">(</span><span class="nam">training_data</span><span class="op">,</span><span class="strut"> </span></p> |
| <p id="t74" class="pln"> <span class="nam">steps_per_epoch</span><span class="op">=</span><span class="nam">params</span><span class="op">[</span><span class="str">'STEPS'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p> |
| <p id="t75" class="pln"> <span class="nam">epochs</span><span class="op">=</span><span class="nam">params</span><span class="op">[</span><span class="str">'EPOCHS'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p> |
| <p id="t76" class="pln"> <span class="nam">validation_data</span><span class="op">=</span><span class="nam">validation_data</span><span class="op">,</span><span class="strut"> </span></p> |
| <p id="t77" class="pln"> <span class="nam">validation_steps</span><span class="op">=</span><span class="nam">params</span><span class="op">[</span><span class="str">'VAL_STEPS'</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p> |
| <p id="t78" class="pln"><span class="strut"> </span></p> |
| <p id="t79" class="stm run hide_run"> <span class="nam">self</span><span class="op">.</span><span class="nam">marvin_model</span> <span class="op">=</span> <span class="nam">model</span><span class="strut"> </span></p> |
| |
| </td> |
| </tr> |
| </table> |
| </div> |
| |
| <div id="footer"> |
| <div class="content"> |
| <p> |
| <a class="nav" href="index.html">« index</a> <a class="nav" href="https://coverage.readthedocs.io">coverage.py v4.5.1</a>, |
| created at 2018-06-27 11:51 |
| </p> |
| </div> |
| </div> |
| |
| </body> |
| </html> |
