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

some maintenance, documentation, and validation improvements

Browse source
This commit is contained in:
peregrinates 2025-08-30 20:18:15 -04:00
parent 975c4f59b9
commit 6db0aa8136
No known key found for this signature in database
GPG key ID: 1CDDD19B6F8CC99D
9 changed files with 570 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

15
analysis/p3_make_phase_data.py
View file

@ -86,7 +86,8 @@ import copy
import re
import argparse
from pathlib import Path
from analysis.analysis_helpers import process_standard_game_inputs, get_country_to_model_mapping, COUNTRIES
from analysis.analysis_helpers import process_standard_game_inputs, get_country_to_model_mapping
from analysis.schemas import COUNTRIES
from tqdm import tqdm
import traceback
@ -94,6 +95,18 @@ def make_phase_data(country_to_model : pd.Series,
lmvs_data : pd.DataFrame,
conversations_data : pd.DataFrame,
orders_data : pd.DataFrame) -> pd.DataFrame:
"""
takes country-to-model mapping, game state (lmvs_data), conversations, and orders, and returns a dataframe with one row per (power, phase).
Args:
country_to_model: mapping of country to model
lmvs_data: raw lmvs_data dataframe
conversations_data: dataframe of conversations
orders_data: dataframe of orders
Returns:
dataframe with one row per (power, phase) containing phase-level features, convos, relationships, and orders info.
"""
longform_conversations_complete = []
for c in COUNTRIES: