손실 함수는 다음과 같이 나는 손실 함수의 RMSE를 정의하고 있습니다

Question

을 int로 :

model.pred= tf.argmax(model.props, 1) 
model.actual = tf.argmax(model.y, 1) 
model.RMSE = tf.sqrt(tf.reduce_mean(tf.square(tf.sub(model.predictedSteer, model.actualSteer)))) 

model.pred은 INT64되지만 입력을 떠 필요로하기 때문에 SQRT는 컴파일 오류를 제공합니다. tf.cast를 사용하면 손실 함수가 차별화되지 않습니다. 이 문제를 어떻게 해결할 수 있습니까?

Answer 1

argmax은 구분할 수 없으므로 관련된 대부분의 기능도 구분할 수 없습니다. 문제에 대해 더 많이 알지 못하면 해결책을 제시하기가 어렵습니다. 분류 문제라면 argmax가 아닌 logits (pred)를 사용하십시오.