2016-10-15 17 views
5

में युग बदलता बनाने के लिए मैं कैसे एक कस्टम नुकसान समारोह एक वजन शब्द है कि क्या युग मैं तैयार हूं के आधार पर अपडेट किया जाता है है कि बनाना चाहेंगेएक नुकसान समारोह जो Keras

उदाहरण के लिए:। मान लीजिए मैं एक नुकसान समारोह जो एक beta वजन, है, जहां पहले 20 अवधियों को अधिक बीटा बढ़ जाती है ...

def custom_loss(x, x_pred): 
    loss1 = objectives.binary_crossentropy(x, x_pred) 
    loss2 = objectives.mse(x, x_pred) 
    return (beta*current_epoch/20) * loss1 + loss2 

मैं एक keras नुकसान समारोह में कुछ इस तरह कैसे लागू कर सकता है?

उत्तर

0

उनके दस्तावेज़ीकरण को देखते हुए वे उल्लेख करते हैं कि आप थैनो/टीएफ प्रतीकात्मक कार्यों का उपयोग कर सकते हैं जो प्रत्येक डेटा बिंदु के लिए एक स्केलर लौटाते हैं। तो आप इस

loss = tf.contrib.losses.softmax_cross_entropy(x, x_pred) * 
     (beta * current_epoch/20) + 
     tf.contrib.losses.mean_squared_error 

की तरह कुछ कर सकता है आप एक्स पारित करने के लिए है और एक्स के रूप में x_pred और tf.placeholders मैं मॉडल बनाने के लिए लगता है कि तुम keras इस्तेमाल कर सकते हैं लेकिन फिर आप चलाना होगा के रूप में x_pred हैं sess.run साथ कम्प्यूटेशनल ग्राफ()

संदर्भ: https://blog.keras.io/keras-as-a-simplified-interface-to-tensorflow-tutorial.html#using-keras-models-with-tensorflow

संबंधित मुद्दे