表格输出实验数据¶
PARL提供了将训练过程中的指标输出到CSV表格的工具。工具导入方法:
from parl.utils import CSVLogger
使用教程¶
传入CSV文件保存路径,并初始化CSVLogger:
csv_logger = CSVLogger("result.csv")
输出以字典形式记录的指标:
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