sklearn_bridge class for easier imports