有谁知道我们如何在DNNClassifier tensorflow premade estimator中更改损失函数？

Question

我想在DNNClassifier中使用单独的损失函数，因为我想要使用的数据非常不平衡 tf.nn.weighted_cross_entropy_with_logits作为损失函数，但我想我需要为它构建一个新的估算器？ 是否可以通过tensorflow Estimator API更改现有预焙DNNC分类器中的损失函数？

Answer 1

您可以在隐藏图层中设置分类器的优化程序和激活功能，但我不认为您可以定义自定义损失功能。

由于您的输入数据是高度不平衡的，＆＃34;您可以通过将权重分配给构造函数的weight_column参数来设置自定义权重。文档为here。