評估器
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 的預測結果。