I\'m training a Keras model which sits in a Scikit pipeline with some preprocessing. The Keras model is defined as
def create_model(X_train): inp = Input(