GitHub – skorch-dev/skorch: A scikit-learn compatible neural network library that wraps pytorch

https://github.com/skorch-dev/skorch

GitHub – skorch-dev/skorch: A scikit-learn compatible neural network library that wraps pytorch
* Some cleanups in test_scoring * indentation level * disable some pylint messages * unused fixtures * Don’t iterate over data when using cached scoring Before, net.infer was cached when using a scoring callback wiht use_caching=True.
github.com

Laisser un commentaire