gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-01 01:45:44 24 4
gpt4 key购买 nike

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

最佳答案

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

由于您的输入数据“高度不平衡”,您可以通过将权重分配给构造函数的 weight_column 来设置自定义权重。争论。文档是 here .

关于tensorflow - 有谁知道我们如何更改 DNNClassifier tensorflow 预制估计器中的损失函数?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49580522/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com