Mercurial > repos > bgruening > sklearn_stacking_ensemble_models
annotate search_model_validation.py @ 10:2d890789ac48 draft
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
| author | bgruening | 
|---|---|
| date | Tue, 13 Apr 2021 21:46:25 +0000 | 
| parents | b8c92e94ac1d | 
| children | 0380f10c4e04 | 
| rev | line source | 
|---|---|
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 1 import argparse | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 2 import collections | 
| 10 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 3 import json | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 4 import os | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 5 import pickle | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 6 import sys | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 7 import warnings | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 8 | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 9 import imblearn | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 10 import joblib | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 11 import numpy as np | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 12 import pandas as pd | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 13 import skrebate | 
| 10 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 14 from galaxy_ml.utils import ( | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 15 clean_params, | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 16 get_cv, | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 17 get_main_estimator, | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 18 get_module, | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 19 get_scoring, | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 20 load_model, | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 21 read_columns, | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 22 SafeEval, | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 23 try_get_attr | 
| 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 24 ) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 25 from scipy.io import mmread | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 26 from sklearn import ( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 27 cluster, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 28 decomposition, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 29 feature_selection, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 30 kernel_approximation, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 31 model_selection, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 32 preprocessing, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 33 ) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 34 from sklearn.exceptions import FitFailedWarning | 
| 10 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 35 from sklearn.model_selection import _search, _validation | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 36 from sklearn.model_selection._validation import _score, cross_validate | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 37 | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 38 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 39 _fit_and_score = try_get_attr("galaxy_ml.model_validations", "_fit_and_score") | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 40 setattr(_search, "_fit_and_score", _fit_and_score) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 41 setattr(_validation, "_fit_and_score", _fit_and_score) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 42 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 43 N_JOBS = int(os.environ.get("GALAXY_SLOTS", 1)) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 44 # handle disk cache | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 45 CACHE_DIR = os.path.join(os.getcwd(), "cached") | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 46 del os | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 47 NON_SEARCHABLE = ("n_jobs", "pre_dispatch", "memory", "_path", "nthread", "callbacks") | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 48 | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 49 | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 50 def _eval_search_params(params_builder): | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 51 search_params = {} | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 52 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 53 for p in params_builder["param_set"]: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 54 search_list = p["sp_list"].strip() | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 55 if search_list == "": | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 56 continue | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 57 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 58 param_name = p["sp_name"] | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 59 if param_name.lower().endswith(NON_SEARCHABLE): | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 60 print("Warning: `%s` is not eligible for search and was " "omitted!" % param_name) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 61 continue | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 62 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 63 if not search_list.startswith(":"): | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 64 safe_eval = SafeEval(load_scipy=True, load_numpy=True) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 65 ev = safe_eval(search_list) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 66 search_params[param_name] = ev | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 67 else: | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 68 # Have `:` before search list, asks for estimator evaluatio | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 69 safe_eval_es = SafeEval(load_estimators=True) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 70 search_list = search_list[1:].strip() | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 71 # TODO maybe add regular express check | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 72 ev = safe_eval_es(search_list) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 73 preprocessings = ( | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 74 preprocessing.StandardScaler(), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 75 preprocessing.Binarizer(), | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 76 preprocessing.MaxAbsScaler(), | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 77 preprocessing.Normalizer(), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 78 preprocessing.MinMaxScaler(), | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 79 preprocessing.PolynomialFeatures(), | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 80 preprocessing.RobustScaler(), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 81 feature_selection.SelectKBest(), | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 82 feature_selection.GenericUnivariateSelect(), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 83 feature_selection.SelectPercentile(), | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 84 feature_selection.SelectFpr(), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 85 feature_selection.SelectFdr(), | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 86 feature_selection.SelectFwe(), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 87 feature_selection.VarianceThreshold(), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 88 decomposition.FactorAnalysis(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 89 decomposition.FastICA(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 90 decomposition.IncrementalPCA(), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 91 decomposition.KernelPCA(random_state=0, n_jobs=N_JOBS), | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 92 decomposition.LatentDirichletAllocation(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 93 decomposition.MiniBatchDictionaryLearning(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 94 decomposition.MiniBatchSparsePCA(random_state=0, n_jobs=N_JOBS), | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 95 decomposition.NMF(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 96 decomposition.PCA(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 97 decomposition.SparsePCA(random_state=0, n_jobs=N_JOBS), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 98 decomposition.TruncatedSVD(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 99 kernel_approximation.Nystroem(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 100 kernel_approximation.RBFSampler(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 101 kernel_approximation.AdditiveChi2Sampler(), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 102 kernel_approximation.SkewedChi2Sampler(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 103 cluster.FeatureAgglomeration(), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 104 skrebate.ReliefF(n_jobs=N_JOBS), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 105 skrebate.SURF(n_jobs=N_JOBS), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 106 skrebate.SURFstar(n_jobs=N_JOBS), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 107 skrebate.MultiSURF(n_jobs=N_JOBS), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 108 skrebate.MultiSURFstar(n_jobs=N_JOBS), | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 109 imblearn.under_sampling.ClusterCentroids(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 110 imblearn.under_sampling.CondensedNearestNeighbour(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 111 imblearn.under_sampling.EditedNearestNeighbours(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 112 imblearn.under_sampling.RepeatedEditedNearestNeighbours(random_state=0, n_jobs=N_JOBS), | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 113 imblearn.under_sampling.AllKNN(random_state=0, n_jobs=N_JOBS), | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 114 imblearn.under_sampling.InstanceHardnessThreshold(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 115 imblearn.under_sampling.NearMiss(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 116 imblearn.under_sampling.NeighbourhoodCleaningRule(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 117 imblearn.under_sampling.OneSidedSelection(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 118 imblearn.under_sampling.RandomUnderSampler(random_state=0), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 119 imblearn.under_sampling.TomekLinks(random_state=0, n_jobs=N_JOBS), | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 120 imblearn.over_sampling.ADASYN(random_state=0, n_jobs=N_JOBS), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 121 imblearn.over_sampling.RandomOverSampler(random_state=0), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 122 imblearn.over_sampling.SMOTE(random_state=0, n_jobs=N_JOBS), | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 123 imblearn.over_sampling.SVMSMOTE(random_state=0, n_jobs=N_JOBS), | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 124 imblearn.over_sampling.BorderlineSMOTE(random_state=0, n_jobs=N_JOBS), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 125 imblearn.over_sampling.SMOTENC(categorical_features=[], random_state=0, n_jobs=N_JOBS), | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 126 imblearn.combine.SMOTEENN(random_state=0), | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 127 imblearn.combine.SMOTETomek(random_state=0), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 128 ) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 129 newlist = [] | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 130 for obj in ev: | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 131 if obj is None: | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 132 newlist.append(None) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 133 elif obj == "all_0": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 134 newlist.extend(preprocessings[0:35]) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 135 elif obj == "sk_prep_all": # no KernalCenter() | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 136 newlist.extend(preprocessings[0:7]) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 137 elif obj == "fs_all": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 138 newlist.extend(preprocessings[7:14]) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 139 elif obj == "decomp_all": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 140 newlist.extend(preprocessings[14:25]) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 141 elif obj == "k_appr_all": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 142 newlist.extend(preprocessings[25:29]) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 143 elif obj == "reb_all": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 144 newlist.extend(preprocessings[30:35]) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 145 elif obj == "imb_all": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 146 newlist.extend(preprocessings[35:54]) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 147 elif type(obj) is int and -1 < obj < len(preprocessings): | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 148 newlist.append(preprocessings[obj]) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 149 elif hasattr(obj, "get_params"): # user uploaded object | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 150 if "n_jobs" in obj.get_params(): | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 151 newlist.append(obj.set_params(n_jobs=N_JOBS)) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 152 else: | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 153 newlist.append(obj) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 154 else: | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 155 sys.exit("Unsupported estimator type: %r" % (obj)) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 156 | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 157 search_params[param_name] = newlist | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 158 | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 159 return search_params | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 160 | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 161 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 162 def _handle_X_y( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 163 estimator, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 164 params, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 165 infile1, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 166 infile2, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 167 loaded_df={}, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 168 ref_seq=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 169 intervals=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 170 targets=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 171 fasta_path=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 172 ): | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 173 """read inputs | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 174 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 175 Params | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 176 ------- | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 177 estimator : estimator object | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 178 params : dict | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 179 Galaxy tool parameter inputs | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 180 infile1 : str | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 181 File path to dataset containing features | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 182 infile2 : str | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 183 File path to dataset containing target values | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 184 loaded_df : dict | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 185 Contains loaded DataFrame objects with file path as keys | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 186 ref_seq : str | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 187 File path to dataset containing genome sequence file | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 188 interval : str | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 189 File path to dataset containing interval file | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 190 targets : str | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 191 File path to dataset compressed target bed file | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 192 fasta_path : str | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 193 File path to dataset containing fasta file | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 194 | 
| 3 
963e449636d3
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ba6a47bdf76bbf4cb276206ac1a8cbf61332fd16"
 bgruening parents: 
2diff
changeset | 195 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 196 Returns | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 197 ------- | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 198 estimator : estimator object after setting new attributes | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 199 X : numpy array | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 200 y : numpy array | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 201 """ | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 202 estimator_params = estimator.get_params() | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 203 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 204 input_type = params["input_options"]["selected_input"] | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 205 # tabular input | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 206 if input_type == "tabular": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 207 header = "infer" if params["input_options"]["header1"] else None | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 208 column_option = params["input_options"]["column_selector_options_1"]["selected_column_selector_option"] | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 209 if column_option in [ | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 210 "by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 211 "all_but_by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 212 "by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 213 "all_but_by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 214 ]: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 215 c = params["input_options"]["column_selector_options_1"]["col1"] | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 216 else: | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 217 c = None | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 218 | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 219 df_key = infile1 + repr(header) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 220 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 221 if df_key in loaded_df: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 222 infile1 = loaded_df[df_key] | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 223 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 224 df = pd.read_csv(infile1, sep="\t", header=header, parse_dates=True) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 225 loaded_df[df_key] = df | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 226 | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 227 X = read_columns(df, c=c, c_option=column_option).astype(float) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 228 # sparse input | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 229 elif input_type == "sparse": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 230 X = mmread(open(infile1, "r")) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 231 | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 232 # fasta_file input | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 233 elif input_type == "seq_fasta": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 234 pyfaidx = get_module("pyfaidx") | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 235 sequences = pyfaidx.Fasta(fasta_path) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 236 n_seqs = len(sequences.keys()) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 237 X = np.arange(n_seqs)[:, np.newaxis] | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 238 for param in estimator_params.keys(): | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 239 if param.endswith("fasta_path"): | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 240 estimator.set_params(**{param: fasta_path}) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 241 break | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 242 else: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 243 raise ValueError( | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 244 "The selected estimator doesn't support " | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 245 "fasta file input! Please consider using " | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 246 "KerasGBatchClassifier with " | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 247 "FastaDNABatchGenerator/FastaProteinBatchGenerator " | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 248 "or having GenomeOneHotEncoder/ProteinOneHotEncoder " | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 249 "in pipeline!" | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 250 ) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 251 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 252 elif input_type == "refseq_and_interval": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 253 path_params = { | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 254 "data_batch_generator__ref_genome_path": ref_seq, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 255 "data_batch_generator__intervals_path": intervals, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 256 "data_batch_generator__target_path": targets, | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 257 } | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 258 estimator.set_params(**path_params) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 259 n_intervals = sum(1 for line in open(intervals)) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 260 X = np.arange(n_intervals)[:, np.newaxis] | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 261 | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 262 # Get target y | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 263 header = "infer" if params["input_options"]["header2"] else None | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 264 column_option = params["input_options"]["column_selector_options_2"]["selected_column_selector_option2"] | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 265 if column_option in [ | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 266 "by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 267 "all_but_by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 268 "by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 269 "all_but_by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 270 ]: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 271 c = params["input_options"]["column_selector_options_2"]["col2"] | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 272 else: | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 273 c = None | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 274 | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 275 df_key = infile2 + repr(header) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 276 if df_key in loaded_df: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 277 infile2 = loaded_df[df_key] | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 278 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 279 infile2 = pd.read_csv(infile2, sep="\t", header=header, parse_dates=True) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 280 loaded_df[df_key] = infile2 | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 281 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 282 y = read_columns(infile2, c=c, c_option=column_option, sep="\t", header=header, parse_dates=True) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 283 if len(y.shape) == 2 and y.shape[1] == 1: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 284 y = y.ravel() | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 285 if input_type == "refseq_and_interval": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 286 estimator.set_params(data_batch_generator__features=y.ravel().tolist()) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 287 y = None | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 288 # end y | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 289 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 290 return estimator, X, y | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 291 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 292 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 293 def _do_outer_cv(searcher, X, y, outer_cv, scoring, error_score="raise", outfile=None): | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 294 """Do outer cross-validation for nested CV | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 295 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 296 Parameters | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 297 ---------- | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 298 searcher : object | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 299 SearchCV object | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 300 X : numpy array | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 301 Containing features | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 302 y : numpy array | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 303 Target values or labels | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 304 outer_cv : int or CV splitter | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 305 Control the cv splitting | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 306 scoring : object | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 307 Scorer | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 308 error_score: str, float or numpy float | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 309 Whether to raise fit error or return an value | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 310 outfile : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 311 File path to store the restuls | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 312 """ | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 313 if error_score == "raise": | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 314 rval = cross_validate( | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 315 searcher, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 316 X, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 317 y, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 318 scoring=scoring, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 319 cv=outer_cv, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 320 n_jobs=N_JOBS, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 321 verbose=0, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 322 error_score=error_score, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 323 ) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 324 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 325 warnings.simplefilter("always", FitFailedWarning) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 326 with warnings.catch_warnings(record=True) as w: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 327 try: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 328 rval = cross_validate( | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 329 searcher, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 330 X, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 331 y, | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 332 scoring=scoring, | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 333 cv=outer_cv, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 334 n_jobs=N_JOBS, | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 335 verbose=0, | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 336 error_score=error_score, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 337 ) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 338 except ValueError: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 339 pass | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 340 for warning in w: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 341 print(repr(warning.message)) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 342 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 343 keys = list(rval.keys()) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 344 for k in keys: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 345 if k.startswith("test"): | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 346 rval["mean_" + k] = np.mean(rval[k]) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 347 rval["std_" + k] = np.std(rval[k]) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 348 if k.endswith("time"): | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 349 rval.pop(k) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 350 rval = pd.DataFrame(rval) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 351 rval = rval[sorted(rval.columns)] | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 352 rval.to_csv(path_or_buf=outfile, sep="\t", header=True, index=False) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 353 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 354 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 355 def _do_train_test_split_val( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 356 searcher, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 357 X, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 358 y, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 359 params, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 360 error_score="raise", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 361 primary_scoring=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 362 groups=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 363 outfile=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 364 ): | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 365 """do train test split, searchCV validates on the train and then use | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 366 the best_estimator_ to evaluate on the test | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 367 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 368 Returns | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 369 -------- | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 370 Fitted SearchCV object | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 371 """ | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 372 train_test_split = try_get_attr("galaxy_ml.model_validations", "train_test_split") | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 373 split_options = params["outer_split"] | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 374 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 375 # splits | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 376 if split_options["shuffle"] == "stratified": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 377 split_options["labels"] = y | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 378 X, X_test, y, y_test = train_test_split(X, y, **split_options) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 379 elif split_options["shuffle"] == "group": | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 380 if groups is None: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 381 raise ValueError("No group based CV option was choosen for " "group shuffle!") | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 382 split_options["labels"] = groups | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 383 if y is None: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 384 X, X_test, groups, _ = train_test_split(X, groups, **split_options) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 385 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 386 X, X_test, y, y_test, groups, _ = train_test_split(X, y, groups, **split_options) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 387 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 388 if split_options["shuffle"] == "None": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 389 split_options["shuffle"] = None | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 390 X, X_test, y, y_test = train_test_split(X, y, **split_options) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 391 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 392 if error_score == "raise": | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 393 searcher.fit(X, y, groups=groups) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 394 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 395 warnings.simplefilter("always", FitFailedWarning) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 396 with warnings.catch_warnings(record=True) as w: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 397 try: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 398 searcher.fit(X, y, groups=groups) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 399 except ValueError: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 400 pass | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 401 for warning in w: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 402 print(repr(warning.message)) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 403 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 404 scorer_ = searcher.scorer_ | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 405 if isinstance(scorer_, collections.Mapping): | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 406 is_multimetric = True | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 407 else: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 408 is_multimetric = False | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 409 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 410 best_estimator_ = getattr(searcher, "best_estimator_") | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 411 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 412 # TODO Solve deep learning models in pipeline | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 413 if best_estimator_.__class__.__name__ == "KerasGBatchClassifier": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 414 test_score = best_estimator_.evaluate(X_test, scorer=scorer_, is_multimetric=is_multimetric) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 415 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 416 test_score = _score(best_estimator_, X_test, y_test, scorer_, is_multimetric=is_multimetric) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 417 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 418 if not is_multimetric: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 419 test_score = {primary_scoring: test_score} | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 420 for key, value in test_score.items(): | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 421 test_score[key] = [value] | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 422 result_df = pd.DataFrame(test_score) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 423 result_df.to_csv(path_or_buf=outfile, sep="\t", header=True, index=False) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 424 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 425 return searcher | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 426 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 427 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 428 def main( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 429 inputs, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 430 infile_estimator, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 431 infile1, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 432 infile2, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 433 outfile_result, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 434 outfile_object=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 435 outfile_weights=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 436 groups=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 437 ref_seq=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 438 intervals=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 439 targets=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 440 fasta_path=None, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 441 ): | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 442 """ | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 443 Parameter | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 444 --------- | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 445 inputs : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 446 File path to galaxy tool parameter | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 447 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 448 infile_estimator : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 449 File path to estimator | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 450 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 451 infile1 : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 452 File path to dataset containing features | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 453 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 454 infile2 : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 455 File path to dataset containing target values | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 456 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 457 outfile_result : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 458 File path to save the results, either cv_results or test result | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 459 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 460 outfile_object : str, optional | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 461 File path to save searchCV object | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 462 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 463 outfile_weights : str, optional | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 464 File path to save model weights | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 465 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 466 groups : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 467 File path to dataset containing groups labels | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 468 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 469 ref_seq : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 470 File path to dataset containing genome sequence file | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 471 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 472 intervals : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 473 File path to dataset containing interval file | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 474 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 475 targets : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 476 File path to dataset compressed target bed file | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 477 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 478 fasta_path : str | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 479 File path to dataset containing fasta file | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 480 """ | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 481 warnings.simplefilter("ignore") | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 482 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 483 # store read dataframe object | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 484 loaded_df = {} | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 485 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 486 with open(inputs, "r") as param_handler: | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 487 params = json.load(param_handler) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 488 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 489 # Override the refit parameter | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 490 params["search_schemes"]["options"]["refit"] = True if params["save"] != "nope" else False | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 491 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 492 with open(infile_estimator, "rb") as estimator_handler: | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 493 estimator = load_model(estimator_handler) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 494 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 495 optimizer = params["search_schemes"]["selected_search_scheme"] | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 496 optimizer = getattr(model_selection, optimizer) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 497 | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 498 # handle gridsearchcv options | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 499 options = params["search_schemes"]["options"] | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 500 | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 501 if groups: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 502 header = "infer" if (options["cv_selector"]["groups_selector"]["header_g"]) else None | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 503 column_option = options["cv_selector"]["groups_selector"]["column_selector_options_g"][ | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 504 "selected_column_selector_option_g" | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 505 ] | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 506 if column_option in [ | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 507 "by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 508 "all_but_by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 509 "by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 510 "all_but_by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 511 ]: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 512 c = options["cv_selector"]["groups_selector"]["column_selector_options_g"]["col_g"] | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 513 else: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 514 c = None | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 515 | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 516 df_key = groups + repr(header) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 517 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 518 groups = pd.read_csv(groups, sep="\t", header=header, parse_dates=True) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 519 loaded_df[df_key] = groups | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 520 | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 521 groups = read_columns( | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 522 groups, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 523 c=c, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 524 c_option=column_option, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 525 sep="\t", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 526 header=header, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 527 parse_dates=True, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 528 ) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 529 groups = groups.ravel() | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 530 options["cv_selector"]["groups_selector"] = groups | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 531 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 532 splitter, groups = get_cv(options.pop("cv_selector")) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 533 options["cv"] = splitter | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 534 primary_scoring = options["scoring"]["primary_scoring"] | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 535 # get_scoring() expects secondary_scoring to be a comma separated string (not a list) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 536 # Check if secondary_scoring is specified | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 537 secondary_scoring = options["scoring"].get("secondary_scoring", None) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 538 if secondary_scoring is not None: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 539 # If secondary_scoring is specified, convert the list into comman separated string | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 540 options["scoring"]["secondary_scoring"] = ",".join(options["scoring"]["secondary_scoring"]) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 541 options["scoring"] = get_scoring(options["scoring"]) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 542 if options["error_score"]: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 543 options["error_score"] = "raise" | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 544 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 545 options["error_score"] = np.NaN | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 546 if options["refit"] and isinstance(options["scoring"], dict): | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 547 options["refit"] = primary_scoring | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 548 if "pre_dispatch" in options and options["pre_dispatch"] == "": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 549 options["pre_dispatch"] = None | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 550 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 551 params_builder = params["search_schemes"]["search_params_builder"] | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 552 param_grid = _eval_search_params(params_builder) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 553 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 554 estimator = clean_params(estimator) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 555 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 556 # save the SearchCV object without fit | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 557 if params["save"] == "save_no_fit": | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 558 searcher = optimizer(estimator, param_grid, **options) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 559 print(searcher) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 560 with open(outfile_object, "wb") as output_handler: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 561 pickle.dump(searcher, output_handler, pickle.HIGHEST_PROTOCOL) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 562 return 0 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 563 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 564 # read inputs and loads new attributes, like paths | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 565 estimator, X, y = _handle_X_y( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 566 estimator, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 567 params, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 568 infile1, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 569 infile2, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 570 loaded_df=loaded_df, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 571 ref_seq=ref_seq, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 572 intervals=intervals, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 573 targets=targets, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 574 fasta_path=fasta_path, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 575 ) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 576 | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 577 # cache iraps_core fits could increase search speed significantly | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 578 memory = joblib.Memory(location=CACHE_DIR, verbose=0) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 579 main_est = get_main_estimator(estimator) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 580 if main_est.__class__.__name__ == "IRAPSClassifier": | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 581 main_est.set_params(memory=memory) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 582 | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 583 searcher = optimizer(estimator, param_grid, **options) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 584 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 585 split_mode = params["outer_split"].pop("split_mode") | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 586 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 587 if split_mode == "nested_cv": | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 588 # make sure refit is choosen | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 589 # this could be True for sklearn models, but not the case for | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 590 # deep learning models | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 591 if not options["refit"] and not all(hasattr(estimator, attr) for attr in ("config", "model_type")): | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 592 warnings.warn("Refit is change to `True` for nested validation!") | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 593 setattr(searcher, "refit", True) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 594 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 595 outer_cv, _ = get_cv(params["outer_split"]["cv_selector"]) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 596 # nested CV, outer cv using cross_validate | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 597 if options["error_score"] == "raise": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 598 rval = cross_validate( | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 599 searcher, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 600 X, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 601 y, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 602 scoring=options["scoring"], | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 603 cv=outer_cv, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 604 n_jobs=N_JOBS, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 605 verbose=options["verbose"], | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 606 return_estimator=(params["save"] == "save_estimator"), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 607 error_score=options["error_score"], | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 608 return_train_score=True, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 609 ) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 610 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 611 warnings.simplefilter("always", FitFailedWarning) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 612 with warnings.catch_warnings(record=True) as w: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 613 try: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 614 rval = cross_validate( | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 615 searcher, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 616 X, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 617 y, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 618 scoring=options["scoring"], | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 619 cv=outer_cv, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 620 n_jobs=N_JOBS, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 621 verbose=options["verbose"], | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 622 return_estimator=(params["save"] == "save_estimator"), | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 623 error_score=options["error_score"], | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 624 return_train_score=True, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 625 ) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 626 except ValueError: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 627 pass | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 628 for warning in w: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 629 print(repr(warning.message)) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 630 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 631 fitted_searchers = rval.pop("estimator", []) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 632 if fitted_searchers: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 633 import os | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 634 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 635 pwd = os.getcwd() | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 636 save_dir = os.path.join(pwd, "cv_results_in_folds") | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 637 try: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 638 os.mkdir(save_dir) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 639 for idx, obj in enumerate(fitted_searchers): | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 640 target_name = "cv_results_" + "_" + "split%d" % idx | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 641 target_path = os.path.join(pwd, save_dir, target_name) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 642 cv_results_ = getattr(obj, "cv_results_", None) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 643 if not cv_results_: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 644 print("%s is not available" % target_name) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 645 continue | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 646 cv_results_ = pd.DataFrame(cv_results_) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 647 cv_results_ = cv_results_[sorted(cv_results_.columns)] | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 648 cv_results_.to_csv(target_path, sep="\t", header=True, index=False) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 649 except Exception as e: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 650 print(e) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 651 finally: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 652 del os | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 653 | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 654 keys = list(rval.keys()) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 655 for k in keys: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 656 if k.startswith("test"): | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 657 rval["mean_" + k] = np.mean(rval[k]) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 658 rval["std_" + k] = np.std(rval[k]) | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 659 if k.endswith("time"): | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 660 rval.pop(k) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 661 rval = pd.DataFrame(rval) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 662 rval = rval[sorted(rval.columns)] | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 663 rval.to_csv(path_or_buf=outfile_result, sep="\t", header=True, index=False) | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 664 # deprecate train test split mode | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 665 """searcher = _do_train_test_split_val( | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 666 searcher, X, y, params, | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 667 primary_scoring=primary_scoring, | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 668 error_score=options['error_score'], | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 669 groups=groups, | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 670 outfile=outfile_result)""" | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 671 return 0 | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 672 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 673 # no outer split | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 674 else: | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 675 searcher.set_params(n_jobs=N_JOBS) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 676 if options["error_score"] == "raise": | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 677 searcher.fit(X, y, groups=groups) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 678 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 679 warnings.simplefilter("always", FitFailedWarning) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 680 with warnings.catch_warnings(record=True) as w: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 681 try: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 682 searcher.fit(X, y, groups=groups) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 683 except ValueError: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 684 pass | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 685 for warning in w: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 686 print(repr(warning.message)) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 687 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 688 cv_results = pd.DataFrame(searcher.cv_results_) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 689 cv_results = cv_results[sorted(cv_results.columns)] | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 690 cv_results.to_csv(path_or_buf=outfile_result, sep="\t", header=True, index=False) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 691 | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 692 memory.clear(warn=False) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 693 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 694 # output best estimator, and weights if applicable | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 695 if outfile_object: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 696 best_estimator_ = getattr(searcher, "best_estimator_", None) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 697 if not best_estimator_: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 698 warnings.warn( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 699 "GridSearchCV object has no attribute " | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 700 "'best_estimator_', because either it's " | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 701 "nested gridsearch or `refit` is False!" | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 702 ) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 703 return | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 704 | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 705 # clean prams | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 706 best_estimator_ = clean_params(best_estimator_) | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 707 | 
| 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 708 main_est = get_main_estimator(best_estimator_) | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 709 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 710 if hasattr(main_est, "model_") and hasattr(main_est, "save_weights"): | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 711 if outfile_weights: | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 712 main_est.save_weights(outfile_weights) | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 713 del main_est.model_ | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 714 del main_est.fit_params | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 715 del main_est.model_class_ | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 716 del main_est.validation_data | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 717 if getattr(main_est, "data_generator_", None): | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 718 del main_est.data_generator_ | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 719 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 720 with open(outfile_object, "wb") as output_handler: | 
| 7 
00819b7f2f55
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
 bgruening parents: 
3diff
changeset | 721 print("Best estimator is saved: %s " % repr(best_estimator_)) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 722 pickle.dump(best_estimator_, output_handler, pickle.HIGHEST_PROTOCOL) | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 723 | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 724 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 725 if __name__ == "__main__": | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 726 aparser = argparse.ArgumentParser() | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 727 aparser.add_argument("-i", "--inputs", dest="inputs", required=True) | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 728 aparser.add_argument("-e", "--estimator", dest="infile_estimator") | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 729 aparser.add_argument("-X", "--infile1", dest="infile1") | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 730 aparser.add_argument("-y", "--infile2", dest="infile2") | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 731 aparser.add_argument("-O", "--outfile_result", dest="outfile_result") | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 732 aparser.add_argument("-o", "--outfile_object", dest="outfile_object") | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 733 aparser.add_argument("-w", "--outfile_weights", dest="outfile_weights") | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 734 aparser.add_argument("-g", "--groups", dest="groups") | 
| 2 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 735 aparser.add_argument("-r", "--ref_seq", dest="ref_seq") | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 736 aparser.add_argument("-b", "--intervals", dest="intervals") | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 737 aparser.add_argument("-t", "--targets", dest="targets") | 
| 
e18d9b17c322
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit c035d399196b3bef9982db4f8e47331411dbb20e
 bgruening parents: 
0diff
changeset | 738 aparser.add_argument("-f", "--fasta_path", dest="fasta_path") | 
| 0 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 739 args = aparser.parse_args() | 
| 
47467890f541
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ab963ec9498bd05d2fb2f24f75adb2fccae7958c
 bgruening parents: diff
changeset | 740 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 741 main( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 742 args.inputs, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 743 args.infile_estimator, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 744 args.infile1, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 745 args.infile2, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 746 args.outfile_result, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 747 outfile_object=args.outfile_object, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 748 outfile_weights=args.outfile_weights, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 749 groups=args.groups, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 750 ref_seq=args.ref_seq, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 751 intervals=args.intervals, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 752 targets=args.targets, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 753 fasta_path=args.fasta_path, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
7diff
changeset | 754 ) | 
