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 
相关文章
相关标签/搜索