Set model as an output in python recipe.

jesus_fm
Level 1
Set model as an output in python recipe.

Hi,

I am new in Dataiku and I would like to know how to save a regressor as an output in a python recipe.

I have seen an option to load a model Interaction with saved models — Dataiku DSS 11 documentation

How can I set as an output the model created for example in this function: 

def train_model(X_train: pd.DataFrame, y_train: pd.Series) -> LinearRegression:
    model = LinearRegression()
    model.fit(X_train, y_train)
    return model

I want to use this model as an input for the test data.

Thanks


Operating system used: Windows 10

0 Kudos
1 Reply
AdrienL
Dataiker

Hi,

If you have a model in python that you want to make into a visual model / Saved model, you can do so via mlflow models.

0 Kudos