我有兩個具體的類,說A和C.我想用一個神經網絡把它們分爲類A,B,C,使得樣本過於接近自信的分類只是歸類爲B 。成本函數應該如下:錯誤分類(A分類爲C,反之亦然)將會有非常大的成本。正確的分類將具有零成本。將項目分類爲B將具有非常低的成本。結果是我們只區分樣本,我們很適合他們各自的類。Tensorflow自定義成本函數
我只有通過TensorFlow簡單的教程的工作,但它並沒有涉及如何定義更具體的成本函數,如這一點。誰能解釋如何能在TensorFlow
完成這裏是我的相關代碼,在這裏我只使用2類分類目前。它是直接從TensorFlow教程:
cost = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(y, y_))
train_step = tf.train.AdamOptimizer(learning_rate = learning_rate).minimize(cost)
y是NN的輸出(將看起來像[[1,0,0],[0,1,0]用於兩個樣品與3設置類),y_是樣本的正確類別,可能是[[1,0,0],[0,0,1]]。在這個例子中,我們會歸入第二樣品B,因爲我們是不確定的,但真正的類是C.
哪裏是你的代碼?請參閱[問]。你甚至不用花費2分鐘的時間[參觀]。 –