GoodStartLabs/AI_Diplomacy
1
0
Fork 0
mirror of https://github.com/GoodStartLabs/AI_Diplomacy.git synced 2026-04-19 12:58:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

state update fixes & streamline prompts

Browse source
This commit is contained in:
sam-paech 2025-07-12 10:17:17 +10:00
parent 1f154a7073
commit b4a56126ec
17 changed files with 710 additions and 281 deletions
Download patch file Download diff file Expand all files Collapse all files

15
experiment_runner/analysis/statistical_game_analysis.py
View file

@ -227,6 +227,8 @@ def _plot_relationships_per_game(
# ── NEW: discard rows with no relationship info ────────────
game_df = game_df[game_df["rel_dict"].apply(bool)]
# ── keep only MOVE phases; drop retreat (R) and adjustment (A) ─────
game_df = game_df[game_df["game_phase"].str.upper().str.endswith("M")]
if game_df.empty: # nothing left to plot
continue
@ -334,9 +336,18 @@ def _plot_relationships_per_game(
else to_rgba(base_colour, alpha=0.35)
)
# ── “double” a lone point so it shows up as a short flat line ──
finite_pts = [(x, y) for x, y in zip(data["x"], y_off) if not math.isnan(y)]
if len(finite_pts) == 1:
x0, y0 = finite_pts[0]
xs = [x0 - 0.05, x0 + 0.05] # tiny horizontal spread
ys = [y0, y0]
else:
xs, ys = data["x"], y_off
plt.plot(
data["x"],
y_off,
xs,
ys,
label=f"{other} ({kind})",
color=colour,
linewidth=2,