跑數據集程序,生成特徵和label需要進行保存,利用到這個函數,生成mat文件。利用MATLAB打開發現label不能顯示,如圖所示,然後整理了一下可能的原因(已解決),下邊是我的猜想,如果有大佬知道,請在下邊留言。
利用io.savemat() 生成一個mat 文件時,不能用tensor,顯示不出來
from scipy import io
import numpy as np
import torch
x = torch.arange(1,22).reshape(7,-1)
y = torch.arange(1,8)
print(y)
y = y.numpy()
print(y)
z = []
for i in range(7):
z.append(y[i]+1)
print('z',z)
result = {'feature':x.numpy(),'label':z}# z 不能是tensor,經過實驗發現可以是numpy或者list類型數據,
io.savemat('SavedData_7',result)
print('finish..........')