关于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)
通过以上步骤,可以顺利转化为张量
标签:gb18030,encoding,张量,转化,train,csv,data 来源: https://blog.csdn.net/huatianxue/article/details/114267041