python scikit-learn wrapper with json-type input features