Set model as an output in python recipe.

jesus_fm
jesus_fm Dataiku DSS Core Designer, Dataiku DSS ML Practitioner, Registered Posts: 1

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

Answers

Setup Info
    Tags
      Help me…