簡體   English   中英

Android中的Tensorflow:線性回歸

[英]Tensorflow in android: Linear regression

我已經使用tensorflow在jupyter筆記本上完成了一個簡單的線性回歸模型的訓練,並且能夠像這樣保存和恢復保存的變量:

抓取數據

現在,我正在嘗試在Android應用程序上使用該模型。

遵循這里的教程,我可以進入導入tensorflow庫的階段,如下所示:

Android JNILibs

現在我要為模型提供輸入數據並獲得輸出值。(請參閱下面的應用程序流程)但是,他們在應用程序中使用了.pb文件(不知道這是什么)。 在4個文件中:

保存的文件

我從保存模型中得到的結果是,我沒有.pb文件,這讓我感到震驚。

應用程序的作用:使用用戶的身高輸入值,使用預訓練的張量流模型預測SoC。 從而,使用線性回歸方程:y = Wx + b

y-SoC

W-重量

x-高度

b-偏見

所有變量均為浮點值。

Android應用流程:

  1. 用戶在文本框中輸入高度值,然后按“預測”按鈕。

  2. 應用程序使用保存的模型的重量,偏差和高度值來預測SoC。

  3. 應用程序在textview中顯示預測的SoC。

所以我的問題是:如何在使用Android Studios 2.3.1的Android應用程序中導入和使用我的模型?

這是我的ipynb和csv數據文件

我可能誤解了這個問題,但是:

鑒於該模型已經過預訓練,權重和偏差不會改變,您可以簡單地使用Jupyter筆記本中計算出的W和b值,並在一個簡單的表達式中對其進行硬編碼

<soc> = -56.0719*<height> + 98.3029

無需為此導入張量流模型。

更新為確保問題得到解答,*。pb文件來自凍結帶有圖形的檢查點文件-有關如何執行此操作,請參閱鏈接教程中的第二個代碼面板。

關於凍結是指這里

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM