python+pytorch實現雙線性模型對鳶尾花數據集訓練

import torch.nn as nnui import torchget from sklearn.datasets import load_irisimport data = load_iris()require   X1 = torch.tensor(data.data).float()float X2 = 0.01 * torch.rand(150, 2).float()im   y 
相關文章
相關標籤/搜索