评估器

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。如果要预测多个张量,则每个张量的预测结果都会被连接起来。