Mercurial > repos > bgruening > sklearn_to_categorical
annotate train_test_eval.py @ 11:caeb57858702 draft default tip
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 57a0433defa3cbc37ab34fbb0ebcfaeb680db8d5
| author | bgruening | 
|---|---|
| date | Sun, 05 Nov 2023 15:19:00 +0000 | 
| parents | b650955a20cc | 
| children | 
| rev | line source | 
|---|---|
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
1 import argparse | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
2 import json | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
3 import os | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
4 import warnings | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
5 from itertools import chain | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
6 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
7 import joblib | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
8 import numpy as np | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
9 import pandas as pd | 
| 
5
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
10 from galaxy_ml.model_persist import dump_model_to_h5, load_model_from_h5 | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
11 from galaxy_ml.model_validations import train_test_split | 
| 
5
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
12 from galaxy_ml.utils import ( | 
| 
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
13 clean_params, | 
| 
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
14 get_module, | 
| 
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
15 get_scoring, | 
| 
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
16 read_columns, | 
| 
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
17 SafeEval, | 
| 
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
18 try_get_attr | 
| 
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
19 ) | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
20 from scipy.io import mmread | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
21 from sklearn import pipeline | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
22 from sklearn.model_selection import _search, _validation | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
23 from sklearn.model_selection._validation import _score | 
| 
5
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
24 from sklearn.utils import _safe_indexing, indexable | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
25 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
26 _fit_and_score = try_get_attr("galaxy_ml.model_validations", "_fit_and_score") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
27 setattr(_search, "_fit_and_score", _fit_and_score) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
28 setattr(_validation, "_fit_and_score", _fit_and_score) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
29 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
30 N_JOBS = int(os.environ.get("GALAXY_SLOTS", 1)) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
31 CACHE_DIR = os.path.join(os.getcwd(), "cached") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
32 del os | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
33 NON_SEARCHABLE = ("n_jobs", "pre_dispatch", "memory", "_path", "nthread", "callbacks") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
34 ALLOWED_CALLBACKS = ( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
35 "EarlyStopping", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
36 "TerminateOnNaN", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
37 "ReduceLROnPlateau", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
38 "CSVLogger", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
39 "None", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
40 ) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
41 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
42 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
43 def _eval_swap_params(params_builder): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
44 swap_params = {} | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
45 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
46 for p in params_builder["param_set"]: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
47 swap_value = p["sp_value"].strip() | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
48 if swap_value == "": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
49 continue | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
50 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
51 param_name = p["sp_name"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
52 if param_name.lower().endswith(NON_SEARCHABLE): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
53 warnings.warn( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
54 "Warning: `%s` is not eligible for search and was " | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
55 "omitted!" % param_name | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
56 ) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
57 continue | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
58 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
59 if not swap_value.startswith(":"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
60 safe_eval = SafeEval(load_scipy=True, load_numpy=True) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
61 ev = safe_eval(swap_value) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
62 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
63 # Have `:` before search list, asks for estimator evaluatio | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
64 safe_eval_es = SafeEval(load_estimators=True) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
65 swap_value = swap_value[1:].strip() | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
66 # TODO maybe add regular express check | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
67 ev = safe_eval_es(swap_value) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
68 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
69 swap_params[param_name] = ev | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
70 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
71 return swap_params | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
72 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
73 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
74 def train_test_split_none(*arrays, **kwargs): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
75 """extend train_test_split to take None arrays | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
76 and support split by group names. | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
77 """ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
78 nones = [] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
79 new_arrays = [] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
80 for idx, arr in enumerate(arrays): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
81 if arr is None: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
82 nones.append(idx) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
83 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
84 new_arrays.append(arr) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
85 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
86 if kwargs["shuffle"] == "None": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
87 kwargs["shuffle"] = None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
88 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
89 group_names = kwargs.pop("group_names", None) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
90 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
91 if group_names is not None and group_names.strip(): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
92 group_names = [name.strip() for name in group_names.split(",")] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
93 new_arrays = indexable(*new_arrays) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
94 groups = kwargs["labels"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
95 n_samples = new_arrays[0].shape[0] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
96 index_arr = np.arange(n_samples) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
97 test = index_arr[np.isin(groups, group_names)] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
98 train = index_arr[~np.isin(groups, group_names)] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
99 rval = list( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
100 chain.from_iterable( | 
| 
5
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
101 (_safe_indexing(a, train), _safe_indexing(a, test)) for a in new_arrays | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
102 ) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
103 ) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
104 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
105 rval = train_test_split(*new_arrays, **kwargs) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
106 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
107 for pos in nones: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
108 rval[pos * 2: 2] = [None, None] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
109 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
110 return rval | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
111 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
112 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
113 def main( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
114 inputs, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
115 infile_estimator, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
116 infile1, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
117 infile2, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
118 outfile_result, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
119 outfile_object=None, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
120 outfile_weights=None, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
121 groups=None, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
122 ref_seq=None, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
123 intervals=None, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
124 targets=None, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
125 fasta_path=None, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
126 ): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
127 """ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
128 Parameter | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
129 --------- | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
130 inputs : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
131 File path to galaxy tool parameter | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
132 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
133 infile_estimator : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
134 File path to estimator | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
135 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
136 infile1 : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
137 File path to dataset containing features | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
138 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
139 infile2 : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
140 File path to dataset containing target values | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
141 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
142 outfile_result : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
143 File path to save the results, either cv_results or test result | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
144 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
145 outfile_object : str, optional | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
146 File path to save searchCV object | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
147 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
148 outfile_weights : str, optional | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
149 File path to save deep learning model weights | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
150 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
151 groups : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
152 File path to dataset containing groups labels | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
153 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
154 ref_seq : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
155 File path to dataset containing genome sequence file | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
156 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
157 intervals : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
158 File path to dataset containing interval file | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
159 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
160 targets : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
161 File path to dataset compressed target bed file | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
162 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
163 fasta_path : str | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
164 File path to dataset containing fasta file | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
165 """ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
166 warnings.simplefilter("ignore") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
167 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
168 with open(inputs, "r") as param_handler: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
169 params = json.load(param_handler) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
170 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
171 # load estimator | 
| 
5
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
172 estimator = load_model_from_h5(infile_estimator) | 
| 
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
173 estimator = clean_params(estimator) | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
174 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
175 # swap hyperparameter | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
176 swapping = params["experiment_schemes"]["hyperparams_swapping"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
177 swap_params = _eval_swap_params(swapping) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
178 estimator.set_params(**swap_params) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
179 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
180 estimator_params = estimator.get_params() | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
181 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
182 # store read dataframe object | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
183 loaded_df = {} | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
184 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
185 input_type = params["input_options"]["selected_input"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
186 # tabular input | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
187 if input_type == "tabular": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
188 header = "infer" if params["input_options"]["header1"] else None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
189 column_option = params["input_options"]["column_selector_options_1"][ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
190 "selected_column_selector_option" | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
191 ] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
192 if column_option in [ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
193 "by_index_number", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
194 "all_but_by_index_number", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
195 "by_header_name", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
196 "all_but_by_header_name", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
197 ]: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
198 c = params["input_options"]["column_selector_options_1"]["col1"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
199 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
200 c = None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
201 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
202 df_key = infile1 + repr(header) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
203 df = pd.read_csv(infile1, sep="\t", header=header, parse_dates=True) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
204 loaded_df[df_key] = df | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
205 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
206 X = read_columns(df, c=c, c_option=column_option).astype(float) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
207 # sparse input | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
208 elif input_type == "sparse": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
209 X = mmread(open(infile1, "r")) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
210 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
211 # fasta_file input | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
212 elif input_type == "seq_fasta": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
213 pyfaidx = get_module("pyfaidx") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
214 sequences = pyfaidx.Fasta(fasta_path) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
215 n_seqs = len(sequences.keys()) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
216 X = np.arange(n_seqs)[:, np.newaxis] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
217 for param in estimator_params.keys(): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
218 if param.endswith("fasta_path"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
219 estimator.set_params(**{param: fasta_path}) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
220 break | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
221 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
222 raise ValueError( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
223 "The selected estimator doesn't support " | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
224 "fasta file input! Please consider using " | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
225 "KerasGBatchClassifier with " | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
226 "FastaDNABatchGenerator/FastaProteinBatchGenerator " | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
227 "or having GenomeOneHotEncoder/ProteinOneHotEncoder " | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
228 "in pipeline!" | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
229 ) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
230 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
231 elif input_type == "refseq_and_interval": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
232 path_params = { | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
233 "data_batch_generator__ref_genome_path": ref_seq, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
234 "data_batch_generator__intervals_path": intervals, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
235 "data_batch_generator__target_path": targets, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
236 } | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
237 estimator.set_params(**path_params) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
238 n_intervals = sum(1 for line in open(intervals)) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
239 X = np.arange(n_intervals)[:, np.newaxis] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
240 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
241 # Get target y | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
242 header = "infer" if params["input_options"]["header2"] else None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
243 column_option = params["input_options"]["column_selector_options_2"][ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
244 "selected_column_selector_option2" | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
245 ] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
246 if column_option in [ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
247 "by_index_number", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
248 "all_but_by_index_number", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
249 "by_header_name", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
250 "all_but_by_header_name", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
251 ]: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
252 c = params["input_options"]["column_selector_options_2"]["col2"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
253 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
254 c = None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
255 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
256 df_key = infile2 + repr(header) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
257 if df_key in loaded_df: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
258 infile2 = loaded_df[df_key] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
259 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
260 infile2 = pd.read_csv(infile2, sep="\t", header=header, parse_dates=True) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
261 loaded_df[df_key] = infile2 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
262 | 
| 
1
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
263 y = read_columns( | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
264 infile2, c=c, c_option=column_option, sep="\t", header=header, parse_dates=True | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
265 ) | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
266 if len(y.shape) == 2 and y.shape[1] == 1: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
267 y = y.ravel() | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
268 if input_type == "refseq_and_interval": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
269 estimator.set_params(data_batch_generator__features=y.ravel().tolist()) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
270 y = None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
271 # end y | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
272 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
273 # load groups | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
274 if groups: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
275 groups_selector = ( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
276 params["experiment_schemes"]["test_split"]["split_algos"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
277 ).pop("groups_selector") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
278 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
279 header = "infer" if groups_selector["header_g"] else None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
280 column_option = groups_selector["column_selector_options_g"][ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
281 "selected_column_selector_option_g" | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
282 ] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
283 if column_option in [ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
284 "by_index_number", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
285 "all_but_by_index_number", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
286 "by_header_name", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
287 "all_but_by_header_name", | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
288 ]: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
289 c = groups_selector["column_selector_options_g"]["col_g"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
290 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
291 c = None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
292 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
293 df_key = groups + repr(header) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
294 if df_key in loaded_df: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
295 groups = loaded_df[df_key] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
296 | 
| 
1
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
297 groups = read_columns( | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
298 groups, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
299 c=c, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
300 c_option=column_option, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
301 sep="\t", | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
302 header=header, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
303 parse_dates=True, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
304 ) | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
305 groups = groups.ravel() | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
306 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
307 # del loaded_df | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
308 del loaded_df | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
309 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
310 # handle memory | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
311 memory = joblib.Memory(location=CACHE_DIR, verbose=0) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
312 # cache iraps_core fits could increase search speed significantly | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
313 if estimator.__class__.__name__ == "IRAPSClassifier": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
314 estimator.set_params(memory=memory) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
315 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
316 # For iraps buried in pipeline | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
317 new_params = {} | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
318 for p, v in estimator_params.items(): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
319 if p.endswith("memory"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
320 # for case of `__irapsclassifier__memory` | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
321 if len(p) > 8 and p[:-8].endswith("irapsclassifier"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
322 # cache iraps_core fits could increase search | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
323 # speed significantly | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
324 new_params[p] = memory | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
325 # security reason, we don't want memory being | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
326 # modified unexpectedly | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
327 elif v: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
328 new_params[p] = None | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
329 # handle n_jobs | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
330 elif p.endswith("n_jobs"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
331 # For now, 1 CPU is suggested for iprasclassifier | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
332 if len(p) > 8 and p[:-8].endswith("irapsclassifier"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
333 new_params[p] = 1 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
334 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
335 new_params[p] = N_JOBS | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
336 # for security reason, types of callback are limited | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
337 elif p.endswith("callbacks"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
338 for cb in v: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
339 cb_type = cb["callback_selection"]["callback_type"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
340 if cb_type not in ALLOWED_CALLBACKS: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
341 raise ValueError("Prohibited callback type: %s!" % cb_type) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
342 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
343 estimator.set_params(**new_params) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
344 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
345 # handle scorer, convert to scorer dict | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
346 # Check if scoring is specified | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
347 scoring = params["experiment_schemes"]["metrics"].get("scoring", None) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
348 if scoring is not None: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
349 # get_scoring() expects secondary_scoring to be a comma separated string (not a list) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
350 # Check if secondary_scoring is specified | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
351 secondary_scoring = scoring.get("secondary_scoring", None) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
352 if secondary_scoring is not None: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
353 # If secondary_scoring is specified, convert the list into comman separated string | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
354 scoring["secondary_scoring"] = ",".join(scoring["secondary_scoring"]) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
355 scorer = get_scoring(scoring) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
356 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
357 # handle test (first) split | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
358 test_split_options = params["experiment_schemes"]["test_split"]["split_algos"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
359 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
360 if test_split_options["shuffle"] == "group": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
361 test_split_options["labels"] = groups | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
362 if test_split_options["shuffle"] == "stratified": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
363 if y is not None: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
364 test_split_options["labels"] = y | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
365 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
366 raise ValueError( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
367 "Stratified shuffle split is not " "applicable on empty target values!" | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
368 ) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
369 | 
| 
1
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
370 ( | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
371 X_train, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
372 X_test, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
373 y_train, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
374 y_test, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
375 groups_train, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
376 _groups_test, | 
| 
 
f93f0cdbaf18
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 
bgruening 
parents: 
0 
diff
changeset
 | 
377 ) = train_test_split_none(X, y, groups, **test_split_options) | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
378 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
379 exp_scheme = params["experiment_schemes"]["selected_exp_scheme"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
380 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
381 # handle validation (second) split | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
382 if exp_scheme == "train_val_test": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
383 val_split_options = params["experiment_schemes"]["val_split"]["split_algos"] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
384 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
385 if val_split_options["shuffle"] == "group": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
386 val_split_options["labels"] = groups_train | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
387 if val_split_options["shuffle"] == "stratified": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
388 if y_train is not None: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
389 val_split_options["labels"] = y_train | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
390 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
391 raise ValueError( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
392 "Stratified shuffle split is not " | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
393 "applicable on empty target values!" | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
394 ) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
395 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
396 ( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
397 X_train, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
398 X_val, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
399 y_train, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
400 y_val, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
401 groups_train, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
402 _groups_val, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
403 ) = train_test_split_none(X_train, y_train, groups_train, **val_split_options) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
404 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
405 # train and eval | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
406 if hasattr(estimator, "validation_data"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
407 if exp_scheme == "train_val_test": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
408 estimator.fit(X_train, y_train, validation_data=(X_val, y_val)) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
409 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
410 estimator.fit(X_train, y_train, validation_data=(X_test, y_test)) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
411 else: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
412 estimator.fit(X_train, y_train) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
413 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
414 if hasattr(estimator, "evaluate"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
415 scores = estimator.evaluate( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
416 X_test, y_test=y_test, scorer=scorer, is_multimetric=True | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
417 ) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
418 else: | 
| 
5
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
419 scores = _score(estimator, X_test, y_test, scorer) | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
420 # handle output | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
421 for name, score in scores.items(): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
422 scores[name] = [score] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
423 df = pd.DataFrame(scores) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
424 df = df[sorted(df.columns)] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
425 df.to_csv(path_or_buf=outfile_result, sep="\t", header=True, index=False) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
426 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
427 memory.clear(warn=False) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
428 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
429 if outfile_object: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
430 main_est = estimator | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
431 if isinstance(estimator, pipeline.Pipeline): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
432 main_est = estimator.steps[-1][-1] | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
433 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
434 if hasattr(main_est, "model_") and hasattr(main_est, "save_weights"): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
435 if outfile_weights: | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
436 main_est.save_weights(outfile_weights) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
437 if getattr(main_est, "model_", None): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
438 del main_est.model_ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
439 if getattr(main_est, "fit_params", None): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
440 del main_est.fit_params | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
441 if getattr(main_est, "model_class_", None): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
442 del main_est.model_class_ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
443 if getattr(main_est, "validation_data", None): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
444 del main_est.validation_data | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
445 if getattr(main_est, "data_generator_", None): | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
446 del main_est.data_generator_ | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
447 | 
| 
5
 
b650955a20cc
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 
bgruening 
parents: 
1 
diff
changeset
 | 
448 dump_model_to_h5(estimator, outfile_object) | 
| 
0
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
449 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
450 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
451 if __name__ == "__main__": | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
452 aparser = argparse.ArgumentParser() | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
453 aparser.add_argument("-i", "--inputs", dest="inputs", required=True) | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
454 aparser.add_argument("-e", "--estimator", dest="infile_estimator") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
455 aparser.add_argument("-X", "--infile1", dest="infile1") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
456 aparser.add_argument("-y", "--infile2", dest="infile2") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
457 aparser.add_argument("-O", "--outfile_result", dest="outfile_result") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
458 aparser.add_argument("-o", "--outfile_object", dest="outfile_object") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
459 aparser.add_argument("-w", "--outfile_weights", dest="outfile_weights") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
460 aparser.add_argument("-g", "--groups", dest="groups") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
461 aparser.add_argument("-r", "--ref_seq", dest="ref_seq") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
462 aparser.add_argument("-b", "--intervals", dest="intervals") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
463 aparser.add_argument("-t", "--targets", dest="targets") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
464 aparser.add_argument("-f", "--fasta_path", dest="fasta_path") | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
465 args = aparser.parse_args() | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
466 | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
467 main( | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
468 args.inputs, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
469 args.infile_estimator, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
470 args.infile1, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
471 args.infile2, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
472 args.outfile_result, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
473 outfile_object=args.outfile_object, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
474 outfile_weights=args.outfile_weights, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
475 groups=args.groups, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
476 ref_seq=args.ref_seq, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
477 intervals=args.intervals, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
478 targets=args.targets, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
479 fasta_path=args.fasta_path, | 
| 
 
59e8b4328c82
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 
bgruening 
parents:  
diff
changeset
 | 
480 ) | 
