/PyTorch-sklearn-with-skorch

Notebook to show the usage of skorch to wrap PyTorch models for making them sklearn compatible.

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers