mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-05-22 14:56:43 +02:00
22 lines
708 B
Python
22 lines
708 B
Python
import pandas as pd
|
|
from pathlib import Path
|
|
import matplotlib.pyplot as plt
|
|
import seaborn as sns
|
|
|
|
dfs = []
|
|
for name in ['mappo']:
|
|
for c in range(5):
|
|
try:
|
|
study_root = Path(__file__).parent / name / f'{name}#{c}'
|
|
print(study_root)
|
|
df = pd.read_csv(study_root / 'results.csv', index_col=False)
|
|
df.reward = df.reward.rolling(100).mean()
|
|
df['method'] = name.upper()
|
|
dfs.append(df)
|
|
except Exception as e:
|
|
pass
|
|
|
|
df = pd.concat(dfs).reset_index()
|
|
sns.lineplot(data=df, x='steps', y='reward', hue='method', palette='husl', ci='sd', linewidth=1.5, err_style='bars')
|
|
plt.savefig('study.png')
|
|
print('saved image') |