search

pytorchbearer / torchbearer

torchbearer: A model fitting library for PyTorch
http://pytorchbearer.org
MIT License
636 stars 65 forks source link

skorch export functionality #76

Closed MattPainter01 closed 5 years ago

ottonemo commented 6 years ago

Maybe you can just provide export functionality so that a model can be imported into skorch which already provides an sklearn interface?

ethanwharris commented 6 years ago

@ottonemo @MattPainter01 skorch export is a nice idea. Cross Val, Grid Seacrh etc. would be very useful. Can just wrap the torchbearer Model in a skorch Net instance so that metrics / callbacks all get called under the hood. Ill look into it :)

ethanwharris commented 5 years ago

This seems unlikely