评估器
tflearn.helpers.evaluator.Evaluator (tensors, model=None, session=None)
用于执行预测和评估模型性能的类。
参数
- tensors:
Tensor
列表。要执行预测的张量列表。 - model:
str
。模型权重路径(可选)。 - session:
Session
。运行预测的会话(可选)。
方法
evaluate (feed_dict, ops, batch_size=128)
评估整个数据集上的一系列张量。通常,'ops' 参数是平均性能指标(例如平均值、前 3 名等)
参数
- feed_dict:
dict
。数据的输入字典。 - ops:
Tensors
列表。要评估的张量。 - batch_size:
int
。批次大小。
返回值
所有批次中每个张量的平均结果。
predict (feed_dict)
通过提供的网络运行数据并返回结果值。
参数
- feed_dict:
dict
。输入数据字典,以占位符作为键,以数据作为值。
返回值
一个 array
。如果要预测多个张量,则每个张量的预测结果都会被连接起来。