Orion
orion.
fit_pipeline
Fit an Orion pipeline to the data.
An str with a path to a JSON file.
str
An str with the name of a registered Orion pipeline.
An MLPipeline instance.
MLPipeline
A dict with an MLPipeline specification.
dict
If no pipeline is passed, the default Orion pipeline is used.
data (str or DataFrame) – Data to which the pipeline should be fitted. It can be passed as a path to a CSV file or as a DataFrame.
pipeline (str, Pipeline or dict) –
An str with the name of a registered pipeline.
hyperparameters (str or dict) – Hyperparameters to set to the pipeline. It can be passed as a hyperparameters dict in the mlblocks format or as a path to the corresponding JSON file. Defaults to None.
mlblocks
None
save_path (str) – Path to the file where the fitted Orion instance will be stored using pickle. If not given, the Orion instance is returned. Defaults to None.
pickle
If no save_path is provided, the fitted Orion instance is returned.