yellowbrick icon indicating copy to clipboard operation
yellowbrick copied to clipboard

Make Neuraxle compatible with Yellowbrick

Open guillaume-chevalier opened this issue 5 years ago • 1 comments

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

  1. Neuraxle steps and pipeline don't directly inherit from sklearn steps, however they almost have the same API
  2. 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 avatar May 24 '20 04:05 guillaume-chevalier

@guillaume-chevalier sounds like a great idea - would welcome a PR in the contrib module for this!

bbengfort avatar Jun 10 '20 13:06 bbengfort