좀 더 자세한 내용을 살펴보기 위해 예제를 진행해 보겠습니다. 먼저 예제를 진행하기 위해 필요한 라이브러리를 호출합니다.
코드 9-27 라이브러리 호출
import pandas as pd
import torch
import torch.nn as nn
from torch.utils.data import Dataset, DataLoader
from sklearn.model_selection import train_test_split ------ 데이터셋을 훈련과 테스트 용도로 분리하기 위한 라이브러리
from sklearn.preprocessing import StandardScaler, MinMaxScaler ------ 정규화와 관련된 라이브러리
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
예제에서 사용할 데이터셋은 당뇨병과 관련된 것입니다. 총 아홉 개의 칼럼으로 구성되어 있으며 여덟 개는 당뇨병을 예측하기 위한 독립 변수, 한 개는 당뇨병인지 아닌지 나타내는 종속 변수로 구성되어 있습니다. 데이터셋은 다음 URL을 통해서도 내려받을 수 있습니다.