TensorFlow分类头的输出设置
2018-09-29 18:14 更新
tf.estimator.export.ClassificationOutput
ClassificationOutput 类
继承自: ExportOutput
定义在:tensorflow/python/estimator/export/export_output.py
表示分类头的输出.
必须设置类、scores 其中的一个,或者两者都设置.
类的张量必须提供字符串标签,而不是整数类 ID.
如果只设置了类,则它被解释为以降序提供前 k 个结果.
如果只设置了scores ,则将其解释为每个类按类 ID 的顺序提供分数.
如果同时设置了类和 scores,则将它们解释为 zip,因此每个 scores 对应于同一索引中的类.客户端不应依赖于条目的顺序.
属性
- classes
- scores
方法
__init__
__init__ ( score = None , classes = None )
构造函数 ClassificationOutput.
分数: 每个类的浮动张量给分数 (有时, 但并不总是解释的概率).可能为 None, 但仅当设置了类.解释不同--请参阅类文档.类: 提供预测类标签的字符串张量.可能是无, 但只有在设置了分数.解释不同--请参阅类文档.提出:
ValueError: 如果没有设置任何类和分数, 或者其中一个不是一个具有正确 dtype 的张量.
ARGS:
- scores:浮动的 Tensor 为每个类提供 scores (有时但并不总是,可以解释为概率).可能为 None,但只有当 classes 被设置时.
- classes:字符串类型的 Tensor 提供预测类标签.可能为 None,但只有当 scores 被设置时.
注意:
- ValueError:如果没有设置类和 scores ,或者其中一个不具有正确 dtype类型的 Tensor.
as_signature_def
as_signature_def ( receiver_tensors )