Mercurial > repos > bgruening > sklearn_stacking_ensemble_models

comparison train_test_eval.py @ 7:00819b7f2f55 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/sklearn commit 756f8be9c3cd437e131e6410cd625c24fe078e8c"
author bgruening
date Wed, 22 Jan 2020 12:33:01 +0000
parents 963e449636d3
children b8c92e94ac1d
comparison
equal deleted inserted replaced
6:aae4725f152b 7:00819b7f2f55
1 import argparse 1 import argparse
2 import joblib 2 import joblib
3 import json 3 import json
4 import numpy as np 4 import numpy as np
5 import os
5 import pandas as pd 6 import pandas as pd
6 import pickle 7 import pickle
7 import warnings 8 import warnings
8 from itertools import chain 9 from itertools import chain
9 from scipy.io import mmread 10 from scipy.io import mmread
27 28
28 _fit_and_score = try_get_attr('galaxy_ml.model_validations', '_fit_and_score') 29 _fit_and_score = try_get_attr('galaxy_ml.model_validations', '_fit_and_score')
29 setattr(_search, '_fit_and_score', _fit_and_score) 30 setattr(_search, '_fit_and_score', _fit_and_score)
30 setattr(_validation, '_fit_and_score', _fit_and_score) 31 setattr(_validation, '_fit_and_score', _fit_and_score)
31 32
32 N_JOBS = int(__import__('os').environ.get('GALAXY_SLOTS', 1)) 33 N_JOBS = int(os.environ.get('GALAXY_SLOTS', 1))
33 CACHE_DIR = './cached' 34 CACHE_DIR = os.path.join(os.getcwd(), 'cached')
35 del os
34 NON_SEARCHABLE = ('n_jobs', 'pre_dispatch', 'memory', '_path', 36 NON_SEARCHABLE = ('n_jobs', 'pre_dispatch', 'memory', '_path',
35 'nthread', 'callbacks') 37 'nthread', 'callbacks')
36 ALLOWED_CALLBACKS = ('EarlyStopping', 'TerminateOnNaN', 'ReduceLROnPlateau', 38 ALLOWED_CALLBACKS = ('EarlyStopping', 'TerminateOnNaN', 'ReduceLROnPlateau',
37 'CSVLogger', 'None') 39 'CSVLogger', 'None')
38 40