評估器

tflearn.helpers.evaluator.Evaluator (tensors, model=None, session=None)

用於執行預測和評估模型效能的類別。

引數

  • tensors: Tensor 的清單。用於執行預測的 Tensor 清單。
  • model: str。模型權重路徑(選用)。
  • session: Session。執行預測的 Session(選用)。

方法

evaluate (feed_dict, ops, batch_size=128)

在整個資料集上評估 Tensor 清單。一般來說,「ops」引數是平均效能指標(例如平均平均值、前 3 名等)

引數
  • feed_dict: dict。資料的 feed 字典。
  • ops: Tensors 的清單。要評估的 Tensor。
  • batch_size: int。批次大小。
回傳值

每個 Tensor 在所有批次中的平均平均結果。

predict (feed_dict)

在提供的網路中執行資料並傳回結果值。

引數
  • feed_dict: dict。Feed 資料字典,其中的佔位符為鍵,而資料為值。
回傳值

array。如果是預測多個 Tensor,則會串接每個 Tensor 的預測結果。