feat(plotting): updated plots to not contain placeholder squares + updated colors
This commit is contained in:
parent
fe7aecb62a
commit
708b061577
3 changed files with 35 additions and 30 deletions
|
|
@ -44,6 +44,7 @@ class Columns(str, Enum):
|
|||
# ... (rest of the file remains same, just need to update plotting functions and obtain_data)
|
||||
"""Column names expected in every evaluation CSV."""
|
||||
|
||||
CHECKPOINT = "checkpoint"
|
||||
ARCH = "architecture"
|
||||
TIMESTEPS = "trained_timesteps"
|
||||
REWARD = "eval_return"
|
||||
|
|
|
|||
Reference in a new issue