iT邦幫忙

2018 iT 邦幫忙鐵人賽
DAY 11
0
AI & Machine Learning

如何使用tensorflow在android上系列 第 11

[Day 11] Tensorflow基本語法介紹(三)

  • 分享至 

  • xImage
  •  

Variable

訓練模型時,通常都會使用變數來儲存或是更新參數,主要都是用在weight跟bias部份,當初使化的時候,要把一個張量當作初始值傳入

import tensorflow as tf
b=tf.Variable(tf.zeros([2]))
init_op=tf.global_variables_initializer()
with tf.Session() as sess:
    sess.run(init_op)
    print(sess.run(b))

以上的tf.global_variables_initializer()這一個函數的用意是對變數進行初始化,然後再透過Session.run()來進行操作,以下就是輸出結果
imgur

train.Saver

會創建一個二進制文件,然後將變數名和tensor值儲存起來,首先要先透過以下的code來建立一個管理模型中的變數的Saver

saver = tf.train.Saver()

然後再透過下列的code把那些變數儲存到相對應的名稱及位置

saver.save(sess, 'model.ckpt')

上一篇
[Day 10] Tensorflow基本語法介紹(二)
下一篇
[Day 12] 介紹激勵函數
系列文
如何使用tensorflow在android上30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言