[英]Tensorflow in android: Linear regression
我已經使用tensorflow在jupyter筆記本上完成了一個簡單的線性回歸模型的訓練,並且能夠像這樣保存和恢復保存的變量:
現在,我正在嘗試在Android應用程序上使用該模型。
遵循這里的教程,我可以進入導入tensorflow庫的階段,如下所示:
現在我要為模型提供輸入數據並獲得輸出值。(請參閱下面的應用程序流程)但是,他們在應用程序中使用了.pb文件(不知道這是什么)。 在4個文件中:
我從保存模型中得到的結果是,我沒有.pb文件,這讓我感到震驚。
應用程序的作用:使用用戶的身高輸入值,使用預訓練的張量流模型預測SoC。 從而,使用線性回歸方程:y = Wx + b
y-SoC
W-重量
x-高度
b-偏見
所有變量均為浮點值。
Android應用流程:
用戶在文本框中輸入高度值,然后按“預測”按鈕。
應用程序使用保存的模型的重量,偏差和高度值來預測SoC。
應用程序在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.