Mercurial > repos > bgruening > keras_train_and_eval
annotate keras_train_and_eval.py @ 16:2af1346e68c9 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:43:41 +0000 | 
| parents | bba502278c25 | 
| children | 
| rev | line source | 
|---|---|
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 1 import argparse | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 2 import json | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 3 import os | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 4 import warnings | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 5 from itertools import chain | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 6 | 
| 5 
785dd890e27d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
4diff
changeset | 7 import joblib | 
| 
785dd890e27d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
4diff
changeset | 8 import numpy as np | 
| 
785dd890e27d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
4diff
changeset | 9 import pandas as pd | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 10 from galaxy_ml.keras_galaxy_models import ( | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 11 _predict_generator, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 12 KerasGBatchClassifier, | 
| 15 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 13 KerasGClassifier, | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 14 KerasGRegressor | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 15 ) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 16 from galaxy_ml.model_persist import dump_model_to_h5, load_model_from_h5 | 
| 5 
785dd890e27d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
4diff
changeset | 17 from galaxy_ml.model_validations import train_test_split | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 18 from galaxy_ml.utils import ( | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 19 clean_params, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 20 gen_compute_scores, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 21 get_main_estimator, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 22 get_module, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 23 get_scoring, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 24 read_columns, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 25 SafeEval | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 26 ) | 
| 5 
785dd890e27d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
4diff
changeset | 27 from scipy.io import mmread | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 28 from sklearn.metrics._scorer import _check_multimetric_scoring | 
| 5 
785dd890e27d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
4diff
changeset | 29 from sklearn.model_selection._validation import _score | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 30 from sklearn.utils import _safe_indexing, indexable | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 31 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 32 N_JOBS = int(os.environ.get("GALAXY_SLOTS", 1)) | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 33 CACHE_DIR = os.path.join(os.getcwd(), "cached") | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 34 NON_SEARCHABLE = ( | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 35 "n_jobs", | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 36 "pre_dispatch", | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 37 "memory", | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 38 "_path", | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 39 "_dir", | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 40 "nthread", | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 41 "callbacks", | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 42 ) | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 43 ALLOWED_CALLBACKS = ( | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 44 "EarlyStopping", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 45 "TerminateOnNaN", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 46 "ReduceLROnPlateau", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 47 "CSVLogger", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 48 "None", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 49 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 50 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 51 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 52 def _eval_swap_params(params_builder): | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 53 swap_params = {} | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 54 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 55 for p in params_builder["param_set"]: | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 56 swap_value = p["sp_value"].strip() | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 57 if swap_value == "": | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 58 continue | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 59 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 60 param_name = p["sp_name"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 61 if param_name.lower().endswith(NON_SEARCHABLE): | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 62 warnings.warn( | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 63 "Warning: `%s` is not eligible for search and was " | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 64 "omitted!" % param_name | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 65 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 66 continue | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 67 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 68 if not swap_value.startswith(":"): | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 69 safe_eval = SafeEval(load_scipy=True, load_numpy=True) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 70 ev = safe_eval(swap_value) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 71 else: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 72 # Have `:` before search list, asks for estimator evaluatio | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 73 safe_eval_es = SafeEval(load_estimators=True) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 74 swap_value = swap_value[1:].strip() | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 75 # TODO maybe add regular express check | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 76 ev = safe_eval_es(swap_value) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 77 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 78 swap_params[param_name] = ev | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 79 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 80 return swap_params | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 81 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 82 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 83 def train_test_split_none(*arrays, **kwargs): | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 84 """extend train_test_split to take None arrays | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 85 and support split by group names. | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 86 """ | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 87 nones = [] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 88 new_arrays = [] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 89 for idx, arr in enumerate(arrays): | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 90 if arr is None: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 91 nones.append(idx) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 92 else: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 93 new_arrays.append(arr) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 94 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 95 if kwargs["shuffle"] == "None": | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 96 kwargs["shuffle"] = None | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 97 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 98 group_names = kwargs.pop("group_names", None) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 99 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 100 if group_names is not None and group_names.strip(): | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 101 group_names = [name.strip() for name in group_names.split(",")] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 102 new_arrays = indexable(*new_arrays) | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 103 groups = kwargs["labels"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 104 n_samples = new_arrays[0].shape[0] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 105 index_arr = np.arange(n_samples) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 106 test = index_arr[np.isin(groups, group_names)] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 107 train = index_arr[~np.isin(groups, group_names)] | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 108 rval = list( | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 109 chain.from_iterable( | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 110 (_safe_indexing(a, train), _safe_indexing(a, test)) for a in new_arrays | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 111 ) | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 112 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 113 else: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 114 rval = train_test_split(*new_arrays, **kwargs) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 115 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 116 for pos in nones: | 
| 5 
785dd890e27d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
4diff
changeset | 117 rval[pos * 2: 2] = [None, None] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 118 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 119 return rval | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 120 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 121 | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 122 def _evaluate_keras_and_sklearn_scores( | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 123 estimator, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 124 data_generator, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 125 X, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 126 y=None, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 127 sk_scoring=None, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 128 steps=None, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 129 batch_size=32, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 130 return_predictions=False, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 131 ): | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 132 """output scores for bother keras and sklearn metrics | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 133 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 134 Parameters | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 135 ----------- | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 136 estimator : object | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 137 Fitted `galaxy_ml.keras_galaxy_models.KerasGBatchClassifier`. | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 138 data_generator : object | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 139 From `galaxy_ml.preprocessors.ImageDataFrameBatchGenerator`. | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 140 X : 2-D array | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 141 Contains indecies of images that need to be evaluated. | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 142 y : None | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 143 Target value. | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 144 sk_scoring : dict | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 145 Galaxy tool input parameters. | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 146 steps : integer or None | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 147 Evaluation/prediction steps before stop. | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 148 batch_size : integer | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 149 Number of samples in a batch | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 150 return_predictions : bool, default is False | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 151 Whether to return predictions and true labels. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 152 """ | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 153 scores = {} | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 154 | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 155 generator = data_generator.flow(X, y=y, batch_size=batch_size) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 156 # keras metrics evaluation | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 157 # handle scorer, convert to scorer dict | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 158 generator.reset() | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 159 score_results = estimator.model_.evaluate_generator(generator, steps=steps) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 160 metrics_names = estimator.model_.metrics_names | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 161 if not isinstance(metrics_names, list): | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 162 scores[metrics_names] = score_results | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 163 else: | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 164 scores = dict(zip(metrics_names, score_results)) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 165 | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 166 if sk_scoring["primary_scoring"] == "default" and not return_predictions: | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 167 return scores | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 168 | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 169 generator.reset() | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 170 predictions, y_true = _predict_generator(estimator.model_, generator, steps=steps) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 171 | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 172 # for sklearn metrics | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 173 if sk_scoring["primary_scoring"] != "default": | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 174 scorer = get_scoring(sk_scoring) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 175 if not isinstance(scorer, (dict, list)): | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 176 scorer = [sk_scoring["primary_scoring"]] | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 177 scorer = _check_multimetric_scoring(estimator, scoring=scorer) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 178 sk_scores = gen_compute_scores(y_true, predictions, scorer) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 179 scores.update(sk_scores) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 180 | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 181 if return_predictions: | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 182 return scores, predictions, y_true | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 183 else: | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 184 return scores, None, None | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 185 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 186 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 187 def main( | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 188 inputs, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 189 infile_estimator, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 190 infile1, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 191 infile2, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 192 outfile_result, | 
| 14 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 193 outfile_history=None, | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 194 outfile_object=None, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 195 outfile_y_true=None, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 196 outfile_y_preds=None, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 197 groups=None, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 198 ref_seq=None, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 199 intervals=None, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 200 targets=None, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 201 fasta_path=None, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 202 ): | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 203 """ | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 204 Parameter | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 205 --------- | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 206 inputs : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 207 File path to galaxy tool parameter. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 208 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 209 infile_estimator : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 210 File path to estimator. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 211 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 212 infile1 : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 213 File path to dataset containing features. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 214 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 215 infile2 : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 216 File path to dataset containing target values. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 217 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 218 outfile_result : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 219 File path to save the results, either cv_results or test result. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 220 | 
| 14 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 221 outfile_history : str, optional | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 222 File path to save the training history. | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 223 | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 224 outfile_object : str, optional | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 225 File path to save searchCV object. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 226 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 227 outfile_y_true : str, optional | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 228 File path to target values for prediction. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 229 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 230 outfile_y_preds : str, optional | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 231 File path to save predictions. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 232 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 233 groups : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 234 File path to dataset containing groups labels. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 235 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 236 ref_seq : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 237 File path to dataset containing genome sequence file. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 238 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 239 intervals : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 240 File path to dataset containing interval file. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 241 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 242 targets : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 243 File path to dataset compressed target bed file. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 244 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 245 fasta_path : str | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 246 File path to dataset containing fasta file. | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 247 """ | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 248 warnings.simplefilter("ignore") | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 249 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 250 with open(inputs, "r") as param_handler: | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 251 params = json.load(param_handler) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 252 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 253 # load estimator | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 254 estimator = load_model_from_h5(infile_estimator) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 255 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 256 estimator = clean_params(estimator) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 257 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 258 # swap hyperparameter | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 259 swapping = params["experiment_schemes"]["hyperparams_swapping"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 260 swap_params = _eval_swap_params(swapping) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 261 estimator.set_params(**swap_params) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 262 estimator_params = estimator.get_params() | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 263 # store read dataframe object | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 264 loaded_df = {} | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 265 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 266 input_type = params["input_options"]["selected_input"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 267 # tabular input | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 268 if input_type == "tabular": | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 269 header = "infer" if params["input_options"]["header1"] else None | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 270 column_option = params["input_options"]["column_selector_options_1"][ | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 271 "selected_column_selector_option" | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 272 ] | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 273 if column_option in [ | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 274 "by_index_number", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 275 "all_but_by_index_number", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 276 "by_header_name", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 277 "all_but_by_header_name", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 278 ]: | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 279 c = params["input_options"]["column_selector_options_1"]["col1"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 280 else: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 281 c = None | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 282 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 283 df_key = infile1 + repr(header) | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 284 df = pd.read_csv(infile1, sep="\t", header=header, parse_dates=True) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 285 loaded_df[df_key] = df | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 286 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 287 X = read_columns(df, c=c, c_option=column_option).astype(float) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 288 # sparse input | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 289 elif input_type == "sparse": | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 290 X = mmread(open(infile1, "r")) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 291 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 292 # fasta_file input | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 293 elif input_type == "seq_fasta": | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 294 pyfaidx = get_module("pyfaidx") | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 295 sequences = pyfaidx.Fasta(fasta_path) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 296 n_seqs = len(sequences.keys()) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 297 X = np.arange(n_seqs)[:, np.newaxis] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 298 for param in estimator_params.keys(): | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 299 if param.endswith("fasta_path"): | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 300 estimator.set_params(**{param: fasta_path}) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 301 break | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 302 else: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 303 raise ValueError( | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 304 "The selected estimator doesn't support " | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 305 "fasta file input! Please consider using " | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 306 "KerasGBatchClassifier with " | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 307 "FastaDNABatchGenerator/FastaProteinBatchGenerator " | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 308 "or having GenomeOneHotEncoder/ProteinOneHotEncoder " | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 309 "in pipeline!" | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 310 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 311 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 312 elif input_type == "refseq_and_interval": | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 313 path_params = { | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 314 "data_batch_generator__ref_genome_path": ref_seq, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 315 "data_batch_generator__intervals_path": intervals, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 316 "data_batch_generator__target_path": targets, | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 317 } | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 318 estimator.set_params(**path_params) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 319 n_intervals = sum(1 for line in open(intervals)) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 320 X = np.arange(n_intervals)[:, np.newaxis] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 321 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 322 # Get target y | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 323 header = "infer" if params["input_options"]["header2"] else None | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 324 column_option = params["input_options"]["column_selector_options_2"][ | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 325 "selected_column_selector_option2" | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 326 ] | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 327 if column_option in [ | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 328 "by_index_number", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 329 "all_but_by_index_number", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 330 "by_header_name", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 331 "all_but_by_header_name", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 332 ]: | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 333 c = params["input_options"]["column_selector_options_2"]["col2"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 334 else: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 335 c = None | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 336 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 337 df_key = infile2 + repr(header) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 338 if df_key in loaded_df: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 339 infile2 = loaded_df[df_key] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 340 else: | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 341 infile2 = pd.read_csv(infile2, sep="\t", header=header, parse_dates=True) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 342 loaded_df[df_key] = infile2 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 343 | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 344 y = read_columns( | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 345 infile2, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 346 c=c, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 347 c_option=column_option, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 348 sep="\t", | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 349 header=header, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 350 parse_dates=True, | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 351 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 352 if len(y.shape) == 2 and y.shape[1] == 1: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 353 y = y.ravel() | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 354 if input_type == "refseq_and_interval": | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 355 estimator.set_params(data_batch_generator__features=y.ravel().tolist()) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 356 y = None | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 357 # end y | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 358 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 359 # load groups | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 360 if groups: | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 361 groups_selector = ( | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 362 params["experiment_schemes"]["test_split"]["split_algos"] | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 363 ).pop("groups_selector") | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 364 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 365 header = "infer" if groups_selector["header_g"] else None | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 366 column_option = groups_selector["column_selector_options_g"][ | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 367 "selected_column_selector_option_g" | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 368 ] | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 369 if column_option in [ | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 370 "by_index_number", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 371 "all_but_by_index_number", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 372 "by_header_name", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 373 "all_but_by_header_name", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 374 ]: | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 375 c = groups_selector["column_selector_options_g"]["col_g"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 376 else: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 377 c = None | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 378 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 379 df_key = groups + repr(header) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 380 if df_key in loaded_df: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 381 groups = loaded_df[df_key] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 382 | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 383 groups = read_columns( | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 384 groups, | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 385 c=c, | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 386 c_option=column_option, | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 387 sep="\t", | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 388 header=header, | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 389 parse_dates=True, | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 390 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 391 groups = groups.ravel() | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 392 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 393 # del loaded_df | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 394 del loaded_df | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 395 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 396 # cache iraps_core fits could increase search speed significantly | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 397 memory = joblib.Memory(location=CACHE_DIR, verbose=0) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 398 main_est = get_main_estimator(estimator) | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 399 if main_est.__class__.__name__ == "IRAPSClassifier": | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 400 main_est.set_params(memory=memory) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 401 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 402 # handle scorer, convert to scorer dict | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 403 scoring = params["experiment_schemes"]["metrics"]["scoring"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 404 scorer = get_scoring(scoring) | 
| 15 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 405 | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 406 # We get 'None' back from the call to 'get_scoring()' if | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 407 # the primary scoring is 'default'. Replace 'default' with | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 408 # the default scoring for classification/regression (accuracy/r2) | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 409 if scorer is None: | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 410 if isinstance(estimator, KerasGClassifier): | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 411 scorer = ['accuracy'] | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 412 if isinstance(estimator, KerasGRegressor): | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 413 scorer = ['r2'] | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 414 | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 415 scorer = _check_multimetric_scoring(estimator, scoring=scorer) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 416 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 417 # handle test (first) split | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 418 test_split_options = params["experiment_schemes"]["test_split"]["split_algos"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 419 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 420 if test_split_options["shuffle"] == "group": | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 421 test_split_options["labels"] = groups | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 422 if test_split_options["shuffle"] == "stratified": | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 423 if y is not None: | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 424 test_split_options["labels"] = y | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 425 else: | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 426 raise ValueError( | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 427 "Stratified shuffle split is not " "applicable on empty target values!" | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 428 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 429 | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 430 X_train, X_test, y_train, y_test, groups_train, groups_test = train_test_split_none( | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 431 X, y, groups, **test_split_options | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 432 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 433 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 434 exp_scheme = params["experiment_schemes"]["selected_exp_scheme"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 435 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 436 # handle validation (second) split | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 437 if exp_scheme == "train_val_test": | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 438 val_split_options = params["experiment_schemes"]["val_split"]["split_algos"] | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 439 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 440 if val_split_options["shuffle"] == "group": | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 441 val_split_options["labels"] = groups_train | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 442 if val_split_options["shuffle"] == "stratified": | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 443 if y_train is not None: | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 444 val_split_options["labels"] = y_train | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 445 else: | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 446 raise ValueError( | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 447 "Stratified shuffle split is not " | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 448 "applicable on empty target values!" | 
| 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 449 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 450 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 451 ( | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 452 X_train, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 453 X_val, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 454 y_train, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 455 y_val, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 456 groups_train, | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 457 groups_val, | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 458 ) = train_test_split_none(X_train, y_train, groups_train, **val_split_options) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 459 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 460 # train and eval | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 461 if hasattr(estimator, "config") and hasattr(estimator, "model_type"): | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 462 if exp_scheme == "train_val_test": | 
| 14 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 463 history = estimator.fit(X_train, y_train, validation_data=(X_val, y_val)) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 464 else: | 
| 14 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 465 history = estimator.fit(X_train, y_train, validation_data=(X_test, y_test)) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 466 else: | 
| 14 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 467 history = estimator.fit(X_train, y_train) | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 468 if "callbacks" in estimator_params: | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 469 for cb in estimator_params["callbacks"]: | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 470 if cb["callback_selection"]["callback_type"] == "CSVLogger": | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 471 hist_df = pd.DataFrame(history.history) | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 472 hist_df["epoch"] = np.arange(1, estimator_params["epochs"] + 1) | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 473 epo_col = hist_df.pop('epoch') | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 474 hist_df.insert(0, 'epoch', epo_col) | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 475 hist_df.to_csv(path_or_buf=outfile_history, sep="\t", header=True, index=False) | 
| 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 476 break | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 477 if isinstance(estimator, KerasGBatchClassifier): | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 478 scores = {} | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 479 steps = estimator.prediction_steps | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 480 batch_size = estimator.batch_size | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 481 data_generator = estimator.data_generator_ | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 482 | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 483 scores, predictions, y_true = _evaluate_keras_and_sklearn_scores( | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 484 estimator, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 485 data_generator, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 486 X_test, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 487 y=y_test, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 488 sk_scoring=scoring, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 489 steps=steps, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 490 batch_size=batch_size, | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 491 return_predictions=bool(outfile_y_true), | 
| 6 
ead8f1822587
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
5diff
changeset | 492 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 493 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 494 else: | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 495 scores = {} | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 496 if hasattr(estimator, "model_") and hasattr(estimator.model_, "metrics_names"): | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 497 batch_size = estimator.batch_size | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 498 score_results = estimator.model_.evaluate( | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 499 X_test, y=y_test, batch_size=batch_size, verbose=0 | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 500 ) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 501 metrics_names = estimator.model_.metrics_names | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 502 if not isinstance(metrics_names, list): | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 503 scores[metrics_names] = score_results | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 504 else: | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 505 scores = dict(zip(metrics_names, score_results)) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 506 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 507 if hasattr(estimator, "predict_proba"): | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 508 predictions = estimator.predict_proba(X_test) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 509 else: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 510 predictions = estimator.predict(X_test) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 511 | 
| 15 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 512 # Un-do OHE of the validation labels | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 513 if len(y_test.shape) == 2: | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 514 rounded_test_labels = np.argmax(y_test, axis=1) | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 515 y_true = rounded_test_labels | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 516 sk_scores = _score(estimator, X_test, rounded_test_labels, scorer) | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 517 else: | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 518 y_true = y_test | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 519 sk_scores = _score(estimator, X_test, y_true, scorer) | 
| 
bba502278c25
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5eca9041ce0154eded5aec07195502d5eb3cdd4f
 bgruening parents: 
14diff
changeset | 520 | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 521 scores.update(sk_scores) | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 522 | 
| 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 523 # handle output | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 524 if outfile_y_true: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 525 try: | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 526 pd.DataFrame(y_true).to_csv(outfile_y_true, sep="\t", index=False) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 527 pd.DataFrame(predictions).astype(np.float32).to_csv( | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 528 outfile_y_preds, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 529 sep="\t", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 530 index=False, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 531 float_format="%g", | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 532 chunksize=10000, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 533 ) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 534 except Exception as e: | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 535 print("Error in saving predictions: %s" % e) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 536 # handle output | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 537 for name, score in scores.items(): | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 538 scores[name] = [score] | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 539 df = pd.DataFrame(scores) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 540 df = df[sorted(df.columns)] | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 541 df.to_csv(path_or_buf=outfile_result, sep="\t", header=True, index=False) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 542 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 543 memory.clear(warn=False) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 544 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 545 if outfile_object: | 
| 10 
b3093f953091
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 9981e25b00de29ed881b2229a173a8c812ded9bb
 bgruening parents: 
6diff
changeset | 546 dump_model_to_h5(estimator, outfile_object) | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 547 | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 548 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 549 if __name__ == "__main__": | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 550 aparser = argparse.ArgumentParser() | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 551 aparser.add_argument("-i", "--inputs", dest="inputs", required=True) | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 552 aparser.add_argument("-e", "--estimator", dest="infile_estimator") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 553 aparser.add_argument("-X", "--infile1", dest="infile1") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 554 aparser.add_argument("-y", "--infile2", dest="infile2") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 555 aparser.add_argument("-O", "--outfile_result", dest="outfile_result") | 
| 14 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 556 aparser.add_argument("-hi", "--outfile_history", dest="outfile_history") | 
| 0 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 557 aparser.add_argument("-o", "--outfile_object", dest="outfile_object") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 558 aparser.add_argument("-l", "--outfile_y_true", dest="outfile_y_true") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 559 aparser.add_argument("-p", "--outfile_y_preds", dest="outfile_y_preds") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 560 aparser.add_argument("-g", "--groups", dest="groups") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 561 aparser.add_argument("-r", "--ref_seq", dest="ref_seq") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 562 aparser.add_argument("-b", "--intervals", dest="intervals") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 563 aparser.add_argument("-t", "--targets", dest="targets") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 564 aparser.add_argument("-f", "--fasta_path", dest="fasta_path") | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 565 args = aparser.parse_args() | 
| 
03f61bb3ca43
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 5b2ac730ec6d3b762faa9034eddd19ad1b347476"
 bgruening parents: diff
changeset | 566 | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 567 main( | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 568 args.inputs, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 569 args.infile_estimator, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 570 args.infile1, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 571 args.infile2, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 572 args.outfile_result, | 
| 14 
818f9b69d8a0
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 80417bf0158a9b596e485dd66408f738f405145a
 bgruening parents: 
10diff
changeset | 573 outfile_history=args.outfile_history, | 
| 4 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 574 outfile_object=args.outfile_object, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 575 outfile_y_true=args.outfile_y_true, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 576 outfile_y_preds=args.outfile_y_preds, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 577 groups=args.groups, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 578 ref_seq=args.ref_seq, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 579 intervals=args.intervals, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 580 targets=args.targets, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 581 fasta_path=args.fasta_path, | 
| 
3866911c93ae
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
0diff
changeset | 582 ) | 
