autobazaar.pipeline module

AutoBazaar Pipeline Module.

Classes

ABPipeline(pipeline_dict, loader, metric, …)

AutoBazaar Pipeline Class.

class autobazaar.pipeline.ABPipeline(pipeline_dict, loader, metric, problem_doc)

Bases: object

AutoBazaar Pipeline Class.

Methods

cv_score(X, y, context[, metric, cv])	Cross Validate this pipeline.
dump(output_dir[, rank])	Dump this pipeline using pickle.
fit(data_params)	Fit the pipeline on the given params.
from_dict(pipeline_dict)	Load a pipeline from a dict.
predict(d3mds)	Get predictions for the given D3MDS.
preprocess(X, y, context)	Execute the preprocessing steps of the pipeline.
to_dict([problem_doc])	Return the details of this pipeline in a dict.

cv_score(X, y, context, metric=None, cv=None)

Cross Validate this pipeline.

dump(output_dir, rank=None)

Dump this pipeline using pickle.

fit(data_params)

Fit the pipeline on the given params.

classmethod from_dict(pipeline_dict)

Load a pipeline from a dict.

predict(d3mds)

Get predictions for the given D3MDS.

preprocess(X, y, context)

Execute the preprocessing steps of the pipeline.

to_dict(problem_doc=False)

Return the details of this pipeline in a dict.