一些函数
np.loadtxt(wine_path, dtype = np.float32, delimiter = ';', skiprows = 1)
,numpy读取csv文件,skiprows跳过第一行。next(csv.reader(open(wine_path), delimiter = ';'))
,next() 返回迭代器的下一个项目,参数为可迭代对象,返回wine_path的第一行。data = wineq[:,:-1]
,去除最后一列数据(评分数据)target = [:,-1]
,取出最后一列数据target_onehot.scatter_(1,target.unsqueeze(1),1.0)
,scatter:将tensor转换成one-hot编码的tensor,广义:将src中的所有值按照index确定的索引写入本tensor中。scatter参数:(dim, index, src),dim指定索引维度,index用来 scatter 的元素索引,src包含要分散的元素或要分散的单个标量的张量。data_min = torch.mean(data,dim = 0)
,求data的平均值,dim=0意为对每一列进行操作,dim=1对每一行进行操作。data_var = torch.var(data, dim = 0)
,求data每列的方差。torch.lt(total_sulfur_data,total_sulfur_threshhold)
,lt小于。