表格输出实验数据

PARL提供了将训练过程中的指标输出到CSV表格的工具。工具导入方法:

from parl.utils import CSVLogger

使用教程

  1. 传入CSV文件保存路径,并初始化CSVLogger:

csv_logger = CSVLogger("result.csv")
  1. 输出以字典形式记录的指标:

Parameters:

  • result(dict) - 需要输出到CSV文件的指标字典

Method:

csv_logger.log_dict({"loss": 1, "reward": 2})

完整例子

from parl.utils import CSVLogger

csv_logger = CSVLogger("result.csv")
csv_logger.log_dict({"loss": 1, "reward": 2})
csv_logger.log_dict({"loss": 3, "reward": 4})

result.csv 文件内容如下:

loss,reward
1,2
3,4