Mercurial > repos > bgruening > create_tool_recommendation_model

diff optimise_hyperparameters.py @ 3:98bc44d17561 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/recommendation_training/tools/tool_recommendation_model commit 65d36f271296a38deeceb0d0e8d471b2898ee8f4"
author bgruening
date Tue, 07 Jul 2020 07:24:21 +0000
parents 50753817983a
children f0da532be419
line wrap: on
line diff
--- a/optimise_hyperparameters.py	Sat May 09 09:38:04 2020 +0000
+++ b/optimise_hyperparameters.py	Tue Jul 07 07:24:21 2020 +0000
@@ -20,7 +20,7 @@
     def __init__(self):
         """ Init method. """
 
-    def train_model(self, config, reverse_dictionary, train_data, train_labels, test_data, test_labels, l_tool_tr_samples, class_weights):
+    def train_model(self, config, reverse_dictionary, train_data, train_labels, test_data, test_labels, tool_tr_samples, class_weights):
         """
         Train a model and report accuracy
         """
@@ -71,7 +71,8 @@
                     train_data,
                     train_labels,
                     batch_size,
-                    l_tool_tr_samples
+                    tool_tr_samples,
+                    reverse_dictionary
                 ),
                 steps_per_epoch=len(train_data) // batch_size,
                 epochs=optimize_n_epochs,