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 |