API Reference

This is the class and function reference of baikal.

Core classes

Step

Mixin class to endow scikit-learn classes with Step capabilities.

Model

A Model is a network (more precisely, a directed acyclic graph) of Steps, and it is defined from the input/output specification of the pipeline.

Steps

Lambda

Step for arbitrary functions.

ColumnStack

Step for stacking arrays along the columns.

Concatenate

Step for concatenating arrays.

Split

Step for splitting arrays.

Stack

Step for stacking arrays.

Utilities

make_step(base_class[, attr_dict, class_name])

Creates a step subclass from the given base class.

plot_model

SKLearnWrapper(build_fn, **params)

Wrapper utility class that allows models to used in scikit-learn’s GridSearchCV API.

get_config()

Get global configuration parameters.

set_config([print_changed_only])

Set global configuration parameters.