Создание конвейера в sklearn с пользовательскими функциями?

Question

Как создать склеарн конвейер с пользовательскими функциями?У меня есть две функции, одна для очистки данных и вторая для построения модели.

def preprocess(df):
   ……………….
   # clean data
   return df_clean

def model(df_clean):
   …………………
   #split data train and test and build randomForest Model
   return model

Поэтому я использую FunctionTransformer и создал конвейер

from sklearn.pipeline import Pipeline, make_pipeline
from sklearn.preprocessing import FunctionTransformer

pipe = Pipeline([("preprocess", FunctionTransformer(preprocess)),("model",FunctionTransformer(model))])

pred = pipe.predict_proba(new_test_data)
print(pred)

Я знаю, что выше неправильно, не знаю какЧтобы продолжить, в канале мне нужно сначала передать данные обучения, затем я должен передать new_test_data?