TensorFlow自定義損失函數——預測商品銷售量

在預測商品銷量時,如果預測多了(預測值比真實銷量大),商家損失的是生產商品的成本;而如果預測少了(預測值比真實銷量小),損失的則是商品的利潤。因爲一般商品的成本和商品的利潤不會嚴格相等,比如如果一個商品的成本是1元,但是利潤是10元,那麼少預測一個就少掙10元;而多預測一個才少掙1元,所以如果神經網絡模型最小化的是均方誤差損失函數,那麼很有可能此模型就無法最大化預期的銷售利潤。

爲了最大化預期利潤,需要將損失函數和利潤直接聯繫起來,需要注意的是,損失函數定義的是損失,所以要將利潤最大化,定義的損失函數應該刻畫成本或者代價,下面的公式給出了一個當預測多於真實值和預測少於真實值時有不同損失係數的損失函數:
在這裏插入圖片描述
其中,yi爲一個batch中第i個數據的真實值,yi’爲神經網絡得到的預測值,a和b是常量,比如在上面介紹的銷量預測問題中,a就等於10 (真實值多於預測值的代價),而b等於1 (真實值少於預測值的代價)。

通過對這個自定義損失函數的優化,模型提供的預測值更有可能最大化收益,在TensorFlow中,可以通過以下代碼來實現這個損失函數:

loss = tf.reduce_sum(tf.where(tf.greater(y_, y), (y_ - y) * loss_less, (y - y_) * loss_more))

①tf.greater函數的輸入是兩個張量,此函數會比較這兩個輸入張量中每一個元素的大小,並返回比較結果,當tf.greater的輸入張量維度不一樣時,TensorFlow會進行類似NumPy廣播操作(broadcasting)的處理;

②tf.where函數有三個參數,第一個爲選擇條件,當選擇條件爲True時,tf.where函數會選擇第二個參數中的值,否則使用第三個參數中的值,需要注意的是,tf.where函數的判斷和選擇都是在元素級別進行的。

接下來使用一段TensorFlow代碼展示這兩個函數的使用:

import tensorflow as tf

v1 = tf.constant([1.0, 2.0, 3.0, 4.0])
v2 = tf.constant([4.0, 3.0, 2.0, 1.0])

with tf.Session() as sess:
    print(sess.run(tf.greater(v1, v2)))
    print(sess.run(tf.where(tf.greater(v1, v2), v1, v2)))
    '''輸出結果爲:
    [False False  True  True]
    [4. 3. 3. 4.]'''

在瞭解如何使用這兩個函數之後,我們來看一看剛纔的預測商品銷售量的實例如何通過具體的TensorFlow代碼實現:

import tensorflow as tf
from numpy.random import RandomState

#聲明wl、W2兩個變量,通過seed參數設定了隨機種子,這樣可以保證每次運行得到的結果是一樣的
w = tf.Variable(tf.random_normal([2, 1], stddev=1, seed=1))

x = tf.placeholder(tf.float32, shape=(None, 2), name="x-input")
y_ = tf.placeholder(tf.float32, shape=(None, 1), name="y-input")

#定義神經網絡結構
y = tf.matmul(x, w)

#定義真實值與預測值之間的交叉熵損失函數,來刻畫真實值與預測值之間的差距
loss_less = 10
loss_more = 1
loss = tf.reduce_sum(tf.where(tf.greater(y_, y), (y_ - y) * loss_less, (y - y_) * loss_more))

#定義反向傳播算法的優化方法
train_step = tf.train.AdamOptimizer(learning_rate=0.001).minimize(loss)

#設置隨機數種子
rdm = RandomState(seed=1)
#設置隨機數據集大小
dataset_size = 128
X = rdm.rand(dataset_size, 2)
'''設置迴歸的正確值爲兩個輸入的和加上一個隨機量。
之所以要加上一個隨機量是爲了加入不可預測的噪音,否則不同損失函數的意義就不大了,因爲不同損失函數都會在能完全預測正確的時候最低。
一般來說噪音爲一個均值爲0的小量,所以這裏的噪音設置爲-0.05——0.05的隨機數。'''
Y = [[x1 + x2 + rdm.rand()/10.0 -0.05] for x1,x2 in X]

#創建會話
with tf.Session() as sess:
    #初始化變量
    init_op = tf.global_variables_initializer()
    sess.run(init_op)
    
    print(sess.run(w))
    
    #設置batch訓練數據的大小
    batch_size = 8
    #設置訓練得輪數
    STEPS = 5000
    for i in range(STEPS):
        #每次選取batch_size個樣本進行訓練
        start = (i * batch_size) % dataset_size
        end = min(start + batch_size, dataset_size)

        #通過選取的樣本訓練神經網絡並更新參數
        sess.run(train_step, feed_dict={x:X[start:end], y_:Y[start:end]})
    
    print(sess.run(w))
    '''輸出結果爲:
    [[-0.8113182]
 	 [ 1.4845988]]
	[[1.019347 ]
 	 [1.0428089]]'''

可以看到參數w優化後,預測函數爲1.019347 * x1 + 1.0428089 * x2,顯然是大於實際的預測函數x1 + x2的,這是因爲我們的損失函數中指定預測少了的損失更大(loss_less > loss_more),所以模型會偏向於預測多一點。

如果我們更換代碼,改爲:

loss_less = 1
loss_more = 10

那麼我們的結果就會變爲:

[[-0.8113182]
 [ 1.4845988]]
[[0.95561105]
 [0.98101896]]

預測函數變爲了0.95561105 * x1 + 0.98101896 * x2,可以看到這時候模型就會偏向於預測少一點。

因此,我們可以得出結論:對於相同的神經網絡,不同的損失函數會對訓練得到的模型產生不同效果。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章