Mercurial > repos > bgruening > sklearn_stacking_ensemble_models
annotate simple_model_fit.py @ 14:2fc88dc854ee draft default tip
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit f031d8ddfb73cec24572648666ac44ee47f08aad
| author | bgruening | 
|---|---|
| date | Thu, 11 Aug 2022 07:49:08 +0000 | 
| parents | 0380f10c4e04 | 
| children | 
| rev | line source | 
|---|---|
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 1 import argparse | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 2 import json | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 3 import pickle | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 4 | 
| 10 
2d890789ac48
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 208a8d348e7c7a182cfbe1b6f17868146428a7e2"
 bgruening parents: 
9diff
changeset | 5 import pandas as pd | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 6 from galaxy_ml.utils import load_model, read_columns | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 7 from scipy.io import mmread | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 8 from sklearn.pipeline import Pipeline | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 9 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 10 N_JOBS = int(__import__("os").environ.get("GALAXY_SLOTS", 1)) | 
| 6 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 11 | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 12 | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 13 # TODO import from galaxy_ml.utils in future versions | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 14 def clean_params(estimator, n_jobs=None): | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 15 """clean unwanted hyperparameter settings | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 16 | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 17 If n_jobs is not None, set it into the estimator, if applicable | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 18 | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 19 Return | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 20 ------ | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 21 Cleaned estimator object | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 22 """ | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 23 ALLOWED_CALLBACKS = ( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 24 "EarlyStopping", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 25 "TerminateOnNaN", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 26 "ReduceLROnPlateau", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 27 "CSVLogger", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 28 "None", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 29 ) | 
| 6 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 30 | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 31 estimator_params = estimator.get_params() | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 32 | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 33 for name, p in estimator_params.items(): | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 34 # all potential unauthorized file write | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 35 if name == "memory" or name.endswith("__memory") or name.endswith("_path"): | 
| 6 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 36 new_p = {name: None} | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 37 estimator.set_params(**new_p) | 
| 11 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 38 elif n_jobs is not None and (name == "n_jobs" or name.endswith("__n_jobs")): | 
| 6 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 39 new_p = {name: n_jobs} | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 40 estimator.set_params(**new_p) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 41 elif name.endswith("callbacks"): | 
| 6 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 42 for cb in p: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 43 cb_type = cb["callback_selection"]["callback_type"] | 
| 6 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 44 if cb_type not in ALLOWED_CALLBACKS: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 45 raise ValueError("Prohibited callback type: %s!" % cb_type) | 
| 6 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 46 | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 47 return estimator | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 48 | 
| 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 49 | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 50 def _get_X_y(params, infile1, infile2): | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 51 """read from inputs and output X and y | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 52 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 53 Parameters | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 54 ---------- | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 55 params : dict | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 56 Tool inputs parameter | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 57 infile1 : str | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 58 File path to dataset containing features | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 59 infile2 : str | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 60 File path to dataset containing target values | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 61 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 62 """ | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 63 # store read dataframe object | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 64 loaded_df = {} | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 65 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 66 input_type = params["input_options"]["selected_input"] | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 67 # tabular input | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 68 if input_type == "tabular": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 69 header = "infer" if params["input_options"]["header1"] else None | 
| 11 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 70 column_option = params["input_options"]["column_selector_options_1"][ | 
| 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 71 "selected_column_selector_option" | 
| 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 72 ] | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 73 if column_option in [ | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 74 "by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 75 "all_but_by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 76 "by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 77 "all_but_by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 78 ]: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 79 c = params["input_options"]["column_selector_options_1"]["col1"] | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 80 else: | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 81 c = None | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 82 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 83 df_key = infile1 + repr(header) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 84 df = pd.read_csv(infile1, sep="\t", header=header, parse_dates=True) | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 85 loaded_df[df_key] = df | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 86 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 87 X = read_columns(df, c=c, c_option=column_option).astype(float) | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 88 # sparse input | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 89 elif input_type == "sparse": | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 90 X = mmread(open(infile1, "r")) | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 91 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 92 # Get target y | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 93 header = "infer" if params["input_options"]["header2"] else None | 
| 11 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 94 column_option = params["input_options"]["column_selector_options_2"][ | 
| 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 95 "selected_column_selector_option2" | 
| 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 96 ] | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 97 if column_option in [ | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 98 "by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 99 "all_but_by_index_number", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 100 "by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 101 "all_but_by_header_name", | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 102 ]: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 103 c = params["input_options"]["column_selector_options_2"]["col2"] | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 104 else: | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 105 c = None | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 106 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 107 df_key = infile2 + repr(header) | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 108 if df_key in loaded_df: | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 109 infile2 = loaded_df[df_key] | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 110 else: | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 111 infile2 = pd.read_csv(infile2, sep="\t", header=header, parse_dates=True) | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 112 loaded_df[df_key] = infile2 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 113 | 
| 11 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 114 y = read_columns( | 
| 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 115 infile2, c=c, c_option=column_option, sep="\t", header=header, parse_dates=True | 
| 
0380f10c4e04
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit ea12f973df4b97a2691d9e4ce6bf6fae59d57717"
 bgruening parents: 
10diff
changeset | 116 ) | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 117 if len(y.shape) == 2 and y.shape[1] == 1: | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 118 y = y.ravel() | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 119 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 120 return X, y | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 121 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 122 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 123 def main(inputs, infile_estimator, infile1, infile2, out_object, out_weights=None): | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 124 """main | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 125 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 126 Parameters | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 127 ---------- | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 128 inputs : str | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 129 File path to galaxy tool parameter | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 130 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 131 infile_estimator : str | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 132 File paths of input estimator | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 133 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 134 infile1 : str | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 135 File path to dataset containing features | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 136 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 137 infile2 : str | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 138 File path to dataset containing target labels | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 139 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 140 out_object : str | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 141 File path for output of fitted model or skeleton | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 142 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 143 out_weights : str | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 144 File path for output of weights | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 145 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 146 """ | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 147 with open(inputs, "r") as param_handler: | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 148 params = json.load(param_handler) | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 149 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 150 # load model | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 151 with open(infile_estimator, "rb") as est_handler: | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 152 estimator = load_model(est_handler) | 
| 6 
aae4725f152b
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit d6333e7294e67be5968a41f404b66699cad4ae53"
 bgruening parents: 
5diff
changeset | 153 estimator = clean_params(estimator, n_jobs=N_JOBS) | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 154 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 155 X_train, y_train = _get_X_y(params, infile1, infile2) | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 156 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 157 estimator.fit(X_train, y_train) | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 158 | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 159 main_est = estimator | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 160 if isinstance(main_est, Pipeline): | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 161 main_est = main_est.steps[-1][-1] | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 162 if hasattr(main_est, "model_") and hasattr(main_est, "save_weights"): | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 163 if out_weights: | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 164 main_est.save_weights(out_weights) | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 165 del main_est.model_ | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 166 del main_est.fit_params | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 167 del main_est.model_class_ | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 168 if getattr(main_est, "validation_data", None): | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 169 del main_est.validation_data | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 170 if getattr(main_est, "data_generator_", None): | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 171 del main_est.data_generator_ | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 172 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 173 with open(out_object, "wb") as output_handler: | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 174 pickle.dump(estimator, output_handler, pickle.HIGHEST_PROTOCOL) | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 175 | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 176 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 177 if __name__ == "__main__": | 
| 5 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 178 aparser = argparse.ArgumentParser() | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 179 aparser.add_argument("-i", "--inputs", dest="inputs", required=True) | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 180 aparser.add_argument("-X", "--infile_estimator", dest="infile_estimator") | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 181 aparser.add_argument("-y", "--infile1", dest="infile1") | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 182 aparser.add_argument("-g", "--infile2", dest="infile2") | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 183 aparser.add_argument("-o", "--out_object", dest="out_object") | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 184 aparser.add_argument("-t", "--out_weights", dest="out_weights") | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 185 args = aparser.parse_args() | 
| 
8b5b653ba1ed
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit eb703290e2589561ea215c84aa9f71bcfe1712c6"
 bgruening parents: diff
changeset | 186 | 
| 9 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 187 main( | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 188 args.inputs, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 189 args.infile_estimator, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 190 args.infile1, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 191 args.infile2, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 192 args.out_object, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 193 args.out_weights, | 
| 
b8c92e94ac1d
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit e2a5eade6d0e5ddf3a47630381a0ad90d80e8a04"
 bgruening parents: 
6diff
changeset | 194 ) | 
