关于csv文件转化为张量

1使用pandas导入

train = pd.read_csv('train_data_head_Chinese.csv', encoding="gb18030")

如果是用的中文的话,记得encoding参数设置为gb18030

2 获取值

pandas是一个数据框,里面也含有表头。我们只关心它的内容,所以,我们使用.values方法来获取
具体操作为:

data = train.values

3 转换为浮点类型数据

data = data.astype(float)

此时,有一个小tips,如果数字里面有许多逗号(千位分隔符),需要进入到csv文件中改一下格式,改为“常规”
在这里插入图片描述

4 转化为张量

y = torch.from_numpy(data)

通过以上步骤,可以顺利转化为张量

猜你喜欢

转载自blog.csdn.net/huatianxue/article/details/114267041
今日推荐