Mark/celeste-ai
Archived
1
0
Fork 0
Code Releases Activity

Fixed minor bug

Browse Source
This commit is contained in:
Mark
2023-02-24 14:24:49 -08:00
parent 19923e672c
commit 4ff32b91ea
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
celeste/plot.py
View File

@ -8,16 +8,13 @@ from multiprocessing import Pool
m = Path("model_data/current")
scaled = True
# Make "predicted reward" plots
def plot_pred(src_model):
plotting.predicted_reward(
src_model,
m / f"plots/predicted/{src_model.stem}.png",
device = torch.device("cpu"),
scaled = scaled
device = torch.device("cpu")
)
# Make "best action" plots
@ -26,8 +23,7 @@ def plot_best(src_model):
src_model,
m / f"plots/best_action/{src_model.stem}.png",
device = torch.device("cpu"),
scaled = scaled
device = torch.device("cpu")
)
# Make "actual reward" plots