diff optimise_hyperparameters.py @ 4:afec8c595124 draft

"planemo upload for repository https://github.com/bgruening/galaxytools/tree/recommendation_training/tools/tool_recommendation_model commit 65d36f271296a38deeceb0d0e8d471b2898ee8f4"

--- a/optimise_hyperparameters.py	Sat May 09 05:38:23 2020 -0400
+++ b/optimise_hyperparameters.py	Tue Jul 07 03:25:49 2020 -0400
@@ -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,