Pipeline sklearn

Hi guys,

How can I build a sklearn pipeline to do the following?

What I have:

A, B = getAB(X_train)
X_train = transform(X_train)
model(A, B, X_train)

What I want:

pipe = Pipeline([
(‘ab’, getAB),
(‘tranf’, transform),
(‘net’, net)

pipe.fit(X_train, y_train)

Please help! :slight_smile:

skorch provides a scikit-learn compatible wrapper to use with your PyTorch model.
Would that work for you?