PyTorch > skorch
A scikit-learn compatible neural network library that wraps PyTorch. <img height="20" src="img/sklearnbig.png" alt="sklearn"> <img height="20" src="img/pytorchbig2.png" alt="PyTorch based/compatible">
A scikit-learn compatible neural network library that wraps PyTorch. <img height="20" src="img/sklearnbig.png" alt="sklearn"> <img height="20" src="img/pytorchbig2.png" alt="PyTorch based/compatible">