oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#1632 - Increased rolling window from 5 to 25

Browse Source
This commit is contained in:
Chris McCarthy
2023-07-20 10:06:44 +01:00
parent afbe2e1400
commit e9fc9a0d1a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
benchmark/primaite_benchmark.py
View File

@@ -112,7 +112,7 @@ def _build_benchmark_latex_report(
doc.append(
f"\nThe mean reward per episode from each session is captured. This is then used to calculate a "
f"combined average reward per episode from the {sessions} individual sessions for smoothing. "
f"Finally, a 5-widow rolling average of the combined average reward per session is calculated for "
f"Finally, a 25-widow rolling average of the combined average reward per session is calculated for "
f"further smoothing."
)
@@ -282,7 +282,7 @@ def _get_df_from_episode_av_reward_dict(data: Dict):
return (
pl.from_dict(data)
.with_columns(rolling_mean=pl.col("av_reward").rolling_mean(window_size=5))
.with_columns(rolling_mean=pl.col("av_reward").rolling_mean(window_size=25))
.rename({"rolling_mean": "rolling_av_reward"})
)