ta2.search module¶

class ta2.search.PipelineSearcher(input_dir='input', output_dir='output', static_dir='static', dump=False, hard_timeout=False, ignore_errors=False, cv_folds=5, subprocess_timeout=None, max_errors=5, store_summary=False)[source]¶

Bases: object

check_stop()[source]¶

get_data_augmentation(dataset, problem)[source]¶

make_btb_scorer(dataset, problem, templates, metric)[source]¶

score_pipeline(dataset, problem, pipeline, metrics=None, random_seed=0, folds=None, stratified=False, shuffle=False, template_name=None)[source]¶

search(dataset, problem, timeout=None, budget=None, templates_csv=None)[source]¶

setup_search()[source]¶

start_session(template_names, dataset, problem, metric, budget)[source]¶

stop()[source]¶

subprocess_evaluate(pipeline, *args, **kwargs)[source]¶

exception ta2.search.ScoringError[source]¶: Bases: Exception

exception ta2.search.SubprocessTimeout[source]¶: Bases: Exception