深度学习:标准化代码

归一化在深度学习训练中非常的重要,可以极大的加快训练速度。
引入scikit-learn的standarScaler来标准化
对训练集进行fit_transform
对测试集进行transform

from sklearn.preprocessing import StandardScaler
train_dataset=scaler.fit_transform(train_dataset)
test_dataset=scaler.transform(test_dataset)

在数据集只有单变量的情况下,要使用np.array.reshape(-1,1)将形状横过来

test_dataset=scaler.transform(np.array(test_dataset).reshape(-1,1))
#归一化完后可以再变回去
test_dataset=test_dataset.reshape(1,-1).squeeze(0)

你可能感兴趣的:(pytorch学习)