Skip to content

joblib.py

Joblib dataclass

Bases: IO[T]

IO for joblib objects.

Example of stand-alone use:

>>> from ordeq_joblib import Joblib
>>> from pathlib import Path
>>> Model = Joblib(path=Path("model.pkl"))
>>> model = Model.load()  # doctest: +SKIP
>>> Model.save(model)  # doctest: +SKIP

Example in a node:

>>> from ordeq_joblib import Joblib
>>> from ordeq_pandas import PandasExcel
>>> import pandas as pd
>>> from pathlib import Path
>>> from ordeq import node
>>> @node(
...     inputs=[
...         Joblib(path=Path("model.pkl")),
...         PandasExcel(path=Path("iris.xlsx"))
...     ],
...     outputs=Joblib(path=Path("model-trained.pkl"))
... )
... def train(model, df):
...     X, y = df.drop('label'), df['label']
...     model.fit(X, y)
...     return model