deploy: db4a7e3

nmcardoso · Aug 28, 2023 · f1c45cc · f1c45cc
1 parent 6236305
commit f1c45cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/_modules/mergernet/estimators/automl.html b/_modules/mergernet/estimators/automl.html
@@ -588,7 +588,7 @@ <h1>Source code for mergernet.estimators.automl</h1><div class="highlight"><pre>
       <span class="n">verbose</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
       <span class="n">save_best_only</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
       <span class="n">save_weights_only</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
-      <span class="n">mode</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">objective_direction</span><span class="p">,</span>
+      <span class="n">mode</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">objective_direction</span><span class="p">[:</span><span class="mi">3</span><span class="p">],</span>
       <span class="n">save_freq</span><span class="o">=</span><span class="s1">&#39;epoch&#39;</span><span class="p">,</span>
       <span class="n">initial_value_threshold</span><span class="o">=</span><span class="n">initial_value</span>
     <span class="p">)</span>