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