tensorflow教程:collection,regularizer
tf.add_to_collection:把變數放入一個集合,把很多變數變成一個列表
tf.get_collection:從一個集合中取出全部變數,是一個列表
tf.add_n:把一個列表的東西都依次加起來
例如:
[python]
- import tensorflow as tf;
- import numpy as np;
- import matplotlib.pyplot as plt;
- v1 = tf.get_variable(name='v1', shape=[1], initializer=tf.constant_initializer(0))
- tf.add_to_collection('loss', v1)
- v2 = tf.get_variable(name='v2', shape=[1], initializer=tf.constant_initializer(2))
- tf.add_to_collection('loss', v2)
- with tf.Session() as sess:
- sess.run(tf.initialize_all_variables())
- print tf.get_collection('loss')
- print sess.run(tf.add_n(tf.get_collection('loss')))
輸出:
[<tensorflow.Python.ops.variables.Variable object at 0x7f6b5d700c50>, <tensorflow.python.ops.variables.Variable object at 0x7f6b5d700c90>]
tensorflow Regularizers
在損失函式上加上正則項是防止過擬合的一個重要方法,下面介紹如何在TensorFlow
中使用正則項.
tensorflow
中對引數使用正則項分為兩步:
1. 建立一個正則方法(函式/物件)
2. 將這個正則方法(函式/物件),應用到引數上
如何建立一個正則方法函式
tf.contrib.layers.l1_regularizer(scale, scope=None)
返回一個用來執行L1
正則化的函式,函式的簽名是func(weights)
.
引數:
- scale: 正則項的係數.
- scope: 可選的
scope name
tf.contrib.layers.l2_regularizer(scale, scope=None)
返回一個執行L2
正則化的函式.
tf.contrib.layers.sum_regularizer(regularizer_list, scope=None)
返回一個可以執行多種(個)正則化的函式.意思是,建立一個正則化方法,這個方法是多個正則化方法的混合體.
引數:
regularizer_list: regulizer的列表
已經知道如何建立正則化方法了,下面要說明的就是如何將正則化方法應用到引數上
應用正則化方法到引數上
tf.contrib.layers.apply_regularization(regularizer, weights_list=None)
先看引數
- regularizer:就是我們上一步建立的正則化方法
- weights_list: 想要執行正則化方法的引數列表,如果為
None
的話,就取GraphKeys.WEIGHTS
中的weights
.
函式返回一個標量Tensor
,同時,這個標量Tensor
也會儲存到GraphKeys.REGULARIZATION_LOSSES
中.這個Tensor
儲存了計算正則項損失的方法.
tensorflow
中的Tensor
是儲存了計算這個值的路徑(方法),當我們run的時候,tensorflow
後端就通過路徑計算出Tensor
對應的值
現在,我們只需將這個正則項損失加到我們的損失函式上就可以了.
如果是自己手動定義
weight
的話,需要手動將weight
儲存到GraphKeys.WEIGHTS
中,但是如果使用layer
的話,就不用這麼麻煩了,別人已經幫你考慮好了.(最好自己驗證一下tf.GraphKeys.WEIGHTS
中是否包含了所有的weights
,防止被坑)
其它
在使用tf.get_variable()
和tf.variable_scope()
的時候,你會發現,它們倆中有regularizer
形參.如果傳入這個引數的話,那麼variable_scope
內的weights
的正則化損失,或者weights
的正則化損失就會被新增到GraphKeys.REGULARIZATION_LOSSES
中.
示例:
import tensorflow as tf
from tensorflow.contrib import layers
regularizer = layers.l1_regularizer(0.1)
with tf.variable_scope('var', initializer=tf.random_normal_initializer(),
regularizer=regularizer):
weight = tf.get_variable('weight', shape=[8], initializer=tf.ones_initializer())
with tf.variable_scope('var2', initializer=tf.random_normal_initializer(),
regularizer=regularizer):
weight2 = tf.get_variable('weight', shape=[8], initializer=tf.ones_initializer())
regularization_loss = tf.reduce_sum(tf.get_collection(tf.GraphKeys.REGULARIZATION_LOSSES))