tf.estimator.export.ClassificationOutput
| Просмотреть исходный код на GitHub |
Представляет вывод головки классификации.
Наследуется от: ExportOutput
tf.estimator.export.ClassificationOutput(
scores=None, classes=None
)
Должны быть заданы либо классы, либо баллы, либо оба.
Классы Tensor должны предоставлять строковые метки, а не целочисленные идентификаторы классов.
Если заданы только классы, это интерпретируется как предоставление результатов топ-k в порядке убывания.
Если заданы только баллы, это интерпретируется как предоставление балла для каждого класса в порядке идентификатора класса.
Если заданы и классы, и баллы, они интерпретируются как скомпонованные, поэтому каждый балл соответствует классу с тем же индексом. Клиенты не должны полагаться на порядок элементов.
| Аргументы | |
|---|---|
scores | Вещественное число Tensor, задающее баллы (иногда, но не всегда интерпретируемые как вероятности) для каждого класса. Может быть None, но только если classes задан. Интерпретация варьируется — см. документацию по классу. |
classes | Строковое Tensor, задающее предсказанные метки классов. Может быть None, но только если scores задан. Интерпретация варьируется — см. документацию по классу. |
| Исключения | |
|---|---|
ValueError | если не заданы ни классы, ни баллы, или один из них не является Tensor с правильным типом данных. |
| Атрибуты | |
|---|---|
classes | |
scores | |
Методы
as_signature_def
as_signature_def(
receiver_tensors
)
Генерация протокола SignatureDef для включения в MetaGraphDef.
SignatureDef будет определять выводы, как описано в этом ExportOutput, и будет использовать указанные receiver_tensors в качестве входных данных.
| Аргументы | |
|---|---|
receiver_tensors | Tensor, или словарь из строк в Tensor, определяющий узлы входных данных, которые будут вводиться. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/estimator/export/ClassificationOutput