ledmands
commited on
Commit
·
44830bc
1
Parent(s):
2ef57f9
Modified plot_improvement.py to fix agent directory sorting bug
Browse files- plot_improvement.py +3 -1
plot_improvement.py
CHANGED
|
@@ -19,7 +19,9 @@ def calc_stats(filepath):
|
|
| 19 |
# args = parser.parse_args()
|
| 20 |
|
| 21 |
filepaths = []
|
| 22 |
-
|
|
|
|
|
|
|
| 23 |
if "dqn_v2" in d:
|
| 24 |
path = "agents/" + d + "/evaluations.npz"
|
| 25 |
filepaths.append(path)
|
|
|
|
| 19 |
# args = parser.parse_args()
|
| 20 |
|
| 21 |
filepaths = []
|
| 22 |
+
agent_dirs = os.listdir("agents/")
|
| 23 |
+
agent_dirs.sort()
|
| 24 |
+
for d in agent_dirs:
|
| 25 |
if "dqn_v2" in d:
|
| 26 |
path = "agents/" + d + "/evaluations.npz"
|
| 27 |
filepaths.append(path)
|