libcity.evaluator.geosan_evaluator¶

class libcity.evaluator.geosan_evaluator.GeoSANEvaluator(config)[源代码]¶

基类：libcity.evaluator.abstract_evaluator.AbstractEvaluator

clear()[源代码]¶: 清除之前收集到的 batch 的评估信息，适用于每次评估开始时进行一次清空，排除之前的评估输入的影响。

collect(batch)[源代码]¶

收集一 batch 的评估输入

参数: batch (torch.Tensor) – 模型输出结果([(1+K)*L, N])

evaluate()[源代码]¶: 返回之前收集到的所有 batch 的评估结果

save_result(save_path, filename=None)[源代码]¶

将评估结果保存到 save_path 文件夹下的 filename 文件中

参数

save_path – 保存路径
filename – 保存文件名