TensorFlow分類頭的輸出設(shè)置

tf.estimator.export.ClassificationOutput

ClassificationOutput 類

繼承自： ExportOutput

定義在：tensorflow/python/estimator/export/export_output.py

表示分類頭的輸出.
必須設(shè)置類、scores 其中的一個(gè),或者兩者都設(shè)置.

類的張量必須提供字符串標(biāo)簽,而不是整數(shù)類 ID.
如果只設(shè)置了類,則它被解釋為以降序提供前 k 個(gè)結(jié)果.

如果只設(shè)置了scores ,則將其解釋為每個(gè)類按類 ID 的順序提供分?jǐn)?shù).

如果同時(shí)設(shè)置了類和 scores,則將它們解釋為 zip,因此每個(gè) scores 對(duì)應(yīng)于同一索引中的類.客戶端不應(yīng)依賴于條目的順序.

屬性

• classes
  
• scores
  

方法

__init__

__init__ (  
    score = None ,  
    classes = None
  )

構(gòu)造函數(shù) ClassificationOutput.

分?jǐn)?shù): 每個(gè)類的浮動(dòng)張量給分?jǐn)?shù) (有時(shí), 但并不總是解釋的概率).可能為 None, 但僅當(dāng)設(shè)置了類.解釋不同--請(qǐng)參閱類文檔.類: 提供預(yù)測(cè)類標(biāo)簽的字符串張量.可能是無, 但只有在設(shè)置了分?jǐn)?shù).解釋不同--請(qǐng)參閱類文檔.提出:
ValueError: 如果沒有設(shè)置任何類和分?jǐn)?shù), 或者其中一個(gè)不是一個(gè)具有正確 dtype 的張量.

ARGS：

• scores：浮動(dòng)的 Tensor 為每個(gè)類提供 scores (有時(shí)但并不總是,可以解釋為概率).可能為 None,但只有當(dāng) classes 被設(shè)置時(shí).
• classes：字符串類型的 Tensor 提供預(yù)測(cè)類標(biāo)簽.可能為 None,但只有當(dāng) scores 被設(shè)置時(shí).

注意：

• ValueError：如果沒有設(shè)置類和 scores ,或者其中一個(gè)不具有正確 dtype類型的 Tensor.

as_signature_def

as_signature_def ( receiver_tensors )