有谁知道我们如何改变 DNNClassifier tensorflow 预制估计器中的损失函数?

Does anyone know how we can change the loss function in DNNClassifier tensorflow premade estimator?

我想在 DNNClassifier 中使用一个单独的损失函数,因为我想使用的数据高度不平衡 tf.nn.weighted_cross_entropy_with_logits 作为损失函数,但我想我需要为它建立一个新的估计器? 是否可以通过 tensorflow Estimator API?

更改现有预烘焙 DNNClassifier 中的损失函数

您可以在隐藏层中设置分类器的优化器和激活函数,但我认为您不能定义自定义损失函数。

由于您的输入数据是 "highly imbalanced,",您可以通过将权重分配给构造函数的 weight_column 参数来设置自定义权重。文档是 here.