목록minibatch (1)

Data Navigator

[PyTorch] Mini Batch and Data Load를 이용한 다중 회귀 분석 모델

In [62]: import numpy as np import torch import torch.nn as nn import torch.nn.functional as F In [63]: from torch.utils.data import TensorDataset from torch.utils.data import DataLoader In [64]: X_train = torch.FloatTensor([[73, 80, 75], [93, 88, 93], [89, 91, 90], [96, 98, 100], [73, 66, 70]]) y_train = torch.FloatTensor([[152], [185], [180], [196], [142]]) In [65]: # TensorDataset은 기본적으로 텐서를 ..
Machine Learning, Deep Learning 2021. 2. 14. 17:41
이전 Prev 1 Next 다음