skorch : Scikit-Learn using Pytorch for NVIDIA GPU

 

Scikit-Learn 은 Machine Learning 분야에서 대표적인 라이브러리임.

https://scikit-learn.org/stable/ 

 

기존 Scikit-Learn은 CPU를 이용한 Parallel Computing이  매우 잘 구현되어 있음.

 

하지만 개인이 사용하기에는 Multi-Core CPU 보다 GPU 1대가 훨씬 비용 대비 성능 증가가 큼.

 

Pytorch는 기존에 CUDA 를 이용하여 NVIDIA GPU를 활용할수있게 개발되어져 있음.

 

Pytorch를 이용하여 Scikit-Learn과 호환되는 프론트엔드를 개발한것이 skorch임

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

 

Deep Learning 을 위한 Neural Network 에 대한 부분만 만들어져 있음

 

이미 훌륭한 분이 좋은 가이드를 남기셧으니 참고함.

https://jehyunlee.github.io/2021/09/29/Python-DL-7-skorch_pipeline/

https://jehyunlee.github.io/2022/06/09/Python-DL-8-skorch_callback/  

https://jehyunlee.github.io/2022/06/09/Python-DL-9-skorch_callback2/    

https://jehyunlee.github.io/2022/06/09/Python-DL-10-skorch_callback3/