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を適用してテンソルを取得します。