Mercurial > repos > bgruening > sklearn_nn_classifier

diff nn_classifier.xml @ 6:e972a913e61a draft
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 8cf3d813ec755166ee0bd517b4ecbbd4f84d4df1
author: bgruening
date: Thu, 23 Aug 2018 16:15:30 -0400
parents: 753ebd417b17
children: 5072ac474cd5
--- a/nn_classifier.xml	Fri Aug 17 12:27:46 2018 -0400
+++ b/nn_classifier.xml	Thu Aug 23 16:15:30 2018 -0400
@@ -19,9 +19,9 @@
 import numpy as np
 import sklearn.neighbors
 import pandas
-import pickle
 
-execfile("$__tool_directory__/utils.py")
+execfile("$__tool_directory__/sk_whitelist.py")
+execfile("$__tool_directory__/utils.py", globals())
 
 input_json_path = sys.argv[1]
 with open(input_json_path, "r") as param_handler:
@@ -30,7 +30,7 @@
 #if $selected_tasks.selected_task == "load":
 
 with open("$infile_model", 'rb') as model_handler:
-    classifier_object = pickle.load(model_handler)
+    classifier_object = SafePickler.load(model_handler)
 
 header = 'infer' if params["selected_tasks"]["header"] else None
 data = pandas.read_csv("$selected_tasks.infile_data", sep='\t', header=header, index_col=None, parse_dates=True, encoding=None, tupleize_cols=False)