Tag: pytorch

pytorch & sklearn pipeline

저는 tabular data를 다룹니다. 간혹 딥러닝을 하고 싶지만 표준화등 전처리도 해야 합니다. 범주형 변수를 인코딩해서 feature importance도 보고 싶습니다. skorch(sklearn + pytorch)를 사용하면 가능합니다. 1. skorch = sklearn + pytorch skorch documentationskorch tu

PyTorch GPU setting

PyTorch GPU 버전을 설치한다고 GPU를 사용하지 않습니다. GPU가 여럿이라면 어떤 GPU가 사용 가능한지 확인하고, 변수와 모델에 GPU 사용 설정을 해야 합니다. 1. PyTorch 버전 확인 현재 버전을 확인합니다.123import torch, torchvisionprint(torch.__version__)print(torchvision.