Mercurial > repos > bgruening > create_tool_recommendation_model
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"
author | bgruening |
---|---|
date | Tue, 07 Jul 2020 03:25:49 -0400 |
parents | 5b3c08710e47 |
children | 4f7e6612906b |
line wrap: on
line diff
--- 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,