Wrap nich classifier in python