浏览代码

Fix numpy warning when stats don't exist (#3251)

/asymm-envs
GitHub 5 年前
当前提交
69e6578d
共有 1 个文件被更改,包括 11 次插入5 次删除
  1. 16
      ml-agents/mlagents/trainers/stats.py

16
ml-agents/mlagents/trainers/stats.py


std: float
num: int
@staticmethod
def empty() -> "StatsSummary":
return StatsSummary(0.0, 0.0, 0)
class StatsWriter(abc.ABC):
"""

:param key: The type of statistic, e.g. Environment/Reward.
:returns: A StatsSummary NamedTuple containing (mean, std, count).
"""
return StatsSummary(
mean=np.mean(StatsReporter.stats_dict[self.category][key]),
std=np.std(StatsReporter.stats_dict[self.category][key]),
num=len(StatsReporter.stats_dict[self.category][key]),
)
if len(StatsReporter.stats_dict[self.category][key]) > 0:
return StatsSummary(
mean=np.mean(StatsReporter.stats_dict[self.category][key]),
std=np.std(StatsReporter.stats_dict[self.category][key]),
num=len(StatsReporter.stats_dict[self.category][key]),
)
return StatsSummary.empty()
正在加载...
取消
保存