Add causal oracle person information screen

This commit is contained in:
Xin Wang
2026-09-07 08:41:34 +08:00
parent fb22108c57
commit ecf00185f1
17 changed files with 1364 additions and 20 deletions

View File

@@ -63,6 +63,7 @@ LAYER_FUSION_MODES = (
)
CURRENT_FRAME_PERSON_SET_FUSION_MODES = (
"frame-set-attention",
"oracle-frame-set-attention",
"audio-person-frame-set",
"gated-person-state-adapter",
)
@@ -75,6 +76,7 @@ LOCAL_FUSION_MODES = ("pooled", *PERSON_SET_FUSION_MODES)
LOCAL_FUSION_SCOPES = {
"pooled": "pooled-baseline",
"frame-set-attention": "f1-current-frame-person-set",
"oracle-frame-set-attention": "causal-oracle-visible-person-speaking-history",
"audio-person-frame-set": "f5-screening-current-frame-no-person-history",
"gated-person-state-adapter": "f6-screening-post-thinker-not-internal",
"track-gru-residual": "causal-anonymous-track-gru-pooled-baseline-residual",
@@ -1303,6 +1305,7 @@ def load_session_data(
sample_id=name,
dataset=dataset,
expected_contract=local_contract,
oracle_analysis=local_fusion == "oracle-frame-set-attention",
)
else:
from local_turn_features import load_local_features