【发布时间】:2017-07-04 09:15:07
【问题描述】:
我已经使用 tensorflow 在 jupyter notebook 上完成了一个简单的线性回归模型的训练,我可以像这样保存和恢复保存的变量:
现在我正在尝试在 android 应用程序上使用该模型。
按照教程here,我可以像这样进入导入tensorflow库的阶段:
现在我想为模型提供输入数据并获得输出值。(请参阅下面的应用程序流程)但是,他们正在使用 .pb 文件(不知道这是什么)他们的申请。在 4 个文件中:
我从保存模型中得到,我没有让我目瞪口呆的 .pb 文件。
应用程序的作用: 使用用户输入的高度值通过预训练的张量流模型预测 SoC。 其中,使用线性回归方程:y = Wx + b
y - SoC
W - 重量
x - 高度
b - 偏见
所有变量都是浮点值。
Android 应用流程:
用户在文本框中输入高度值,然后按下“预测”按钮。
应用程序使用已保存模型的权重、偏差和高度值来预测 SoC。
应用程序在文本视图中显示预测的 SoC。
所以我的问题是:如何使用 android studios 2.3.1 在 Android 应用程序中导入和使用我的模型?
这是我的 ipynb 和 csv 数据files。
【问题讨论】:
-
有关冻结模型以创建 pb 图的信息在您在问题中链接的教程中...在教程的第一部分中,它准确地解释了如何将图和检查点冻结到一个模型。
标签: android python tensorflow linear-regression nightly-build