首页 > 解决方案 > java中的标准标量,如numpy

问题描述

我正在使用 tensorflow 将预先训练的模型加载到 android 中

在我的原始模型中,每当我试图预测某些东西时,我需要使用StandardScaler

from sklearn.preprocessing import StandardScaler
sc = StandardScaler()

model.predict(sc.transform(np.array([[6,5,2,5]])))

在我的 android 部分,我将输入作为浮点数组

float[] input = new float[]{6,5,2,5};

哪个显示不准确的结果,但如果我把

 float[] input = new float[]{-0.45480626,  0.00226552 ,-1.23735555 ,-0.17786957};

它显示准确的结果

这里

print(sc.transform(np.array([[6,5,2,5]])));
>>[[-0.45480626  0.00226552 -1.23735555 -0.17786957]]

如何在 java 中使用像标准标量一样的 numpy

标签: javanumpytensorflow

解决方案


推荐阅读