딥러닝 과제 1 - 순방향 신경망 약물 안전성 예측 pytorch
import torch import torch.nn as nn import torch.optim as optim import numpy as np import torch.nn.init as init def load_data(filename): data = np.loadtxt(filename, delimiter=',', dtype=np.float32) x = data[:, 0:-2] y = data[:, [-1]] # after 3 months (-2) and 6 months (-1) return torch.tensor(x, dtype=torch.float32), torch.tensor(y, dtype=torch.float32) 이번에는 함수로 해서 model을 만들어서 한번 잘게 잘게 보겠습니다. 저는 ..