我已经使用 tensorflow 在 jupyter notebook 上完成了一个简单的线性回归模型的训练,并且我能够像这样保存和恢复保存的变量:
现在我正在尝试在 android 应用程序上使用该模型。
按照此处的教程,我可以进入导入 tensorflow 库的阶段,如下所示:
现在我想给模型一个输入数据并获得一个输出值。(请参阅下面的应用程序流程)但是,他们在他们的应用程序中使用了一个 .pb 文件(不知道这是什么)。在 4 个文件中:
我从保存模型中得到的,我没有让我目瞪口呆的 .pb 文件。
应用程序的作用:使用用户输入的高度值通过预训练的 TensorFlow 模型预测 SoC。因此,使用线性回归方程:y = Wx + b
y - SoC
W - 重量
x - 高度
b - 偏差
所有变量都是浮点值。
安卓应用流程:
用户在文本框中输入高度值,然后按“预测”按钮。
应用程序使用已保存模型的重量、偏差和高度值来预测 SoC。
应用程序在文本视图中显示预测的 SoC。
所以我的问题是:如何使用 android studios 2.3.1 在 Android 应用程序中导入和使用我的模型?
这是我的 ipynb 和 csv 数据文件。