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( doc.append(
f"\nThe mean reward per episode from each session is captured. This is then used to calculate a " 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"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." f"further smoothing."
) )
@@ -282,7 +282,7 @@ def _get_df_from_episode_av_reward_dict(data: Dict):
return ( return (
pl.from_dict(data) 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"}) .rename({"rolling_mean": "rolling_av_reward"})
) )