深度学习：标准化代码
·
归一化在深度学习训练中非常的重要，可以极大的加快训练速度。
引入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)
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
0
0- 0
已为社区贡献2条内容
所有评论(0)