Keras

2016-10-15 11 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

문서를 보면 각 데이터 요소에 대해 스칼라를 반환하는 theano/Tf 기호 함수를 사용할 수 있다고 언급했습니다. 그래서이

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

처럼 뭔가를 할 수 당신은 X를 통과해야하고 X로 x_pred 내가 너희를 keras를 사용할 수있는 모델 생성을위한 생각하지만 다시는 실행해야 tf.placeholders로 x_pred 것 sess.run 그래프로 계산()

참조 : https://blog.keras.io/keras-as-a-simplified-interface-to-tensorflow-tutorial.html#using-keras-models-with-tensorflow