Mark
/
celeste-ai
Archived
1
0
Fork 0

Fixed minor bug

master
Mark 2023-02-24 14:24:49 -08:00
parent 19923e672c
commit 4ff32b91ea
Signed by: Mark
GPG Key ID: AD62BB059C2AAEE4
1 changed files with 2 additions and 6 deletions

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