Iris数据集神经网络训练
Iris数据集中简单来说有四个输入量(四种花的性状),一个输出量(花的种类),神经网络的作用其实就是通过数据集训练来预测未被训练的数据的输出
import csv,pandas as pd
import torch
import numpy as np
import torch.nn as nn
import torch.nn.functional as F
import matplotlib.pyplot as plt
from torch.autograd import Variable
input,output = [],[]#输入输出分别对应列表
with open('Iris数据集下载之后的绝对路径')as(f):
f_csv=csv.reader(f)
headers=next(f_csv)
for row in f_csv:
input.append([row[1],row[2],row[3],row[4]])
#4维张量
output.append([row[5]])
#单输出