yellowbrick
yellowbrick copied to clipboard
Make Neuraxle compatible with Yellowbrick
Describe the solution you'd like It would be cool to create a NeuraxleWrapper that would wrap any Neuraxle step or pipeline, just like how Neuraxle adapts scikit-learn steps into its core using SKLearnWrapper.
Is your feature request related to a problem? Please describe. Related:
- https://github.com/Neuraxio/Neuraxle/issues/340
- https://github.com/Neuraxio/Neuraxle/issues/342 (this suggested solution would avoid you to have to add Neuraxle to your requirements.txt if you do the same)
Examples
- Neuraxle steps and pipeline don't directly inherit from sklearn steps, however they almost have the same API
- Neuraxle steps and pipeline don't have a score method. Thus the NeuraxleWrapper could have such a method, being optionnally provided a metric to use at the construction of the wrapper or with a set_metric method on the wrapper.
@alxkolm
@guillaume-chevalier sounds like a great idea - would welcome a PR in the contrib module for this!