add post-processing in sklearn pipeline
add post-processing in sklearn pipeline