TwoQuarks commited on
Commit
bcd8676
·
verified ·
1 Parent(s): b79e083

Update Down/down/exp/plot_paradox_results.py

Browse files
Down/down/exp/plot_paradox_results.py CHANGED
@@ -19,9 +19,9 @@ def _load_csv(path: Path) -> List[Dict[str, str]]:
19
 
20
 
21
  def main() -> None:
22
- quark_dir = Path(__file__).resolve().parents[1] # .../down
23
- results_dir = Path(os.environ.get("TWOQUARKS_RESULTS_DIR", (quark_dir / "results").as_posix()))
24
- graphics_dir = Path(os.environ.get("TWOQUARKS_GRAPHICS_DIR", (quark_dir / "graphics").as_posix()))
25
  results_dir.mkdir(parents=True, exist_ok=True)
26
  graphics_dir.mkdir(parents=True, exist_ok=True)
27
  results_csv = results_dir / "down_paradox_tabular_results.csv"
@@ -56,7 +56,7 @@ def main() -> None:
56
  plt.title("DOWN: reward over training")
57
  plt.legend()
58
  out1 = graphics_dir / "down_paradox_tabular_reward.png"
59
- plt.savefig(out1, dpi=160, bbox_inches="tight")
60
  plt.close()
61
 
62
  # Rho plot (if present)
@@ -72,7 +72,7 @@ def main() -> None:
72
  plt.title("DOWN: rho over training")
73
  plt.legend()
74
  out2 = graphics_dir / "down_paradox_tabular_rho.png"
75
- plt.savefig(out2, dpi=160, bbox_inches="tight")
76
  plt.close()
77
 
78
  print(f"Saved plots to {graphics_dir}")
 
19
 
20
 
21
  def main() -> None:
22
+ Down_dir = Path(__file__).resolve().parents[1] # .../down
23
+ results_dir = Path(os.environ.get("DOWN_RESULTS_DIR", (Down_dir / "results").as_posix()))
24
+ graphics_dir = Path(os.environ.get("DOWN_GRAPHICS_DIR", (Down_dir / "graphics").as_posix()))
25
  results_dir.mkdir(parents=True, exist_ok=True)
26
  graphics_dir.mkdir(parents=True, exist_ok=True)
27
  results_csv = results_dir / "down_paradox_tabular_results.csv"
 
56
  plt.title("DOWN: reward over training")
57
  plt.legend()
58
  out1 = graphics_dir / "down_paradox_tabular_reward.png"
59
+ plt.savefig(out1, dpi=120, bbox_inches="tight")
60
  plt.close()
61
 
62
  # Rho plot (if present)
 
72
  plt.title("DOWN: rho over training")
73
  plt.legend()
74
  out2 = graphics_dir / "down_paradox_tabular_rho.png"
75
+ plt.savefig(out2, dpi=120, bbox_inches="tight")
76
  plt.close()
77
 
78
  print(f"Saved plots to {graphics_dir}")