Tensorflow Estimator : 가중치 적용 예제 교육

Question

Tensorflow Estimator을 사용하여 모델을 교육 중이며 데이터가 균형을 이루지 않았습니다. 나는 각 훈련 예를 가중함으로써 이것을 교정하고 싶다.

원시 Tensorflow에서 like this 할 수도 있습니다. Estimator에서이 작업을 수행하는 쉬운 방법이 있습니까? 아마도 사용자 정의 input_fn을 작성 하시겠습니까?

Answer 1

당신이 분류를하고 있다고 가정합니다. 그렇다면, tf.estimator.DNNClassifier를 사용

weight_column : 문자열 또는 tf.feature_column.numeric_column이 가중치를 나타내는 기능 열을 정의하여 만든 _NumericColumn합니다. 훈련 도중 체중을 줄이거 나 사례를 높이는 데 사용됩니다. 예제의 손실이 곱해집니다. 문자열 인 경우 features에서 가중치 텐서를 가져 오는 키로 사용됩니다. _NumericColumn 인 경우 원시 텐서를 weight_column.key 키로 가져온 다음 weight_column.normalizer_fn을 적용하여 가중치 텐서를 얻습니다.