Fixed minor bug
parent
19923e672c
commit
4ff32b91ea
|
@ -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
|
||||
|
|
Reference in New Issue