Skorch : Utiliser l'API Scikit-learn sur les modèles PyTorch

skorch est une bibliothèque de haut niveau pour PyTorch qui offre une compatibilité totale avec Scikit-learn. En d'autres termes, elle combine la puissance de PyTorch avec l'élégance de sklearn.

Ainsi, tu peux entraîner des modèles PyTorch d'une manière similaire à Scikit-learn, en utilisant des fonctions telles que fit, predict, score, etc.

En utilisant skorch, tu peux également mettre un modèle PyTorch dans le pipeline sklearn, et bien d'autres choses encore.

Globalement, le but est d'être aussi flexible que PyTorch tout en ayant une interface propre comme sklearn.

En savoir plus : Documentation skorch.

Pour plus de détails du code de l’illustration ci-dessus : Utilisation de skorch pour utiliser l’API de Sckit-Learn.