Spaces:
Running
Running
File size: 6,285 Bytes
3dbcd77 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | """Relationship controls; analysis snapshots are scoped to model and tensor."""
import numpy as np
import streamlit as st
from relationships import analyse_pairs
def render_relationship_controls(vectors, label, scope, save_calculation):
options = list(vectors)
active_key = f'active_relationship:{scope}'
example_key = f'relationship_examples:{scope}'
def capture(key):
return {'key': key, 'label': label(key), 'values': np.array(vectors[key], dtype=float, copy=True)}
def choose(text, suffix, index=0, optional=False):
choices = [None] + options if optional else options
return st.selectbox(text, choices, index=0 if optional else min(index, len(choices) - 1),
format_func=lambda key: 'None' if key is None else label(key),
key=f'rel_{suffix}:{scope}')
def remove_example(identifier):
st.session_state[example_key] = [p for p in st.session_state[example_key] if p['id'] != identifier]
st.caption('A → B is to C → D. Show relationship applies these inputs to the chart.')
if len(options) < 2:
st.info('Add at least two tokens or calculated vectors to the chart to choose a relationship.')
return st.session_state.get(active_key)
mode = st.selectbox('Relationship task', ['Compare relationships', 'Complete analogy', 'Average example pairs'],
key=f'rel_task:{scope}')
if mode == 'Average example pairs':
example_a = choose('Example A', 'example_a')
example_b = choose('Example B', 'example_b', 1)
pairs = st.session_state.setdefault(example_key, [])
if st.button('Add example pair', key='add_relationship_example', disabled=len(pairs) >= 10):
serial = st.session_state.get('relationship_example_serial', 0) + 1
st.session_state.relationship_example_serial = serial
pairs.append({'id': serial, 'a': capture(example_a), 'b': capture(example_b)})
for number, pair in enumerate(pairs, 1):
text, action = st.columns([3, 1.2], vertical_alignment='center')
text.text(f'{number}. {pair["a"]["label"]} → {pair["b"]["label"]}')
action.button('Remove', key=f'remove_relationship_example:{pair["id"]}',
on_click=remove_example, args=(pair['id'],))
st.caption(f'{len(pairs)} / 10 example pairs · offsets are averaged equally')
else:
source_a = choose('A', 'a')
source_b = choose('B', 'b', 1)
pairs = [{'a': capture(source_a), 'b': capture(source_b)}]
target = choose('C (apply relationship to)', 'c', 2)
expected = choose('D (compare against)' if mode == 'Compare relationships' else 'Expected D (optional)',
'd' if mode == 'Compare relationships' else 'expected', 3,
optional=mode != 'Compare relationships')
if st.button('Show relationship', key='show_relationship', type='primary', disabled=not pairs, use_container_width=True):
try:
target_data = capture(target)
expected_data = capture(expected) if expected is not None else None
metrics = analyse_pairs([(p['a']['values'], p['b']['values']) for p in pairs],
target_data['values'], expected_data['values'] if expected_data else None)
serial = st.session_state.get('relationship_serial', 0) + 1
st.session_state.relationship_serial = serial
excluded = {item['key'] for p in pairs for item in [p['a'], p['b']] if isinstance(item['key'], int)}
if isinstance(target, int):
excluded.add(target)
st.session_state[active_key] = {'id': serial, 'mode': mode, 'pairs': list(pairs),
'target': target_data, 'expected': expected_data,
'metrics': metrics, 'exclude_ids': sorted(excluded)}
st.session_state.pending_chart_view = 'Relationship 3D'
st.rerun()
except ValueError as e:
st.error(str(e))
analysis = st.session_state.get(active_key)
if analysis is None:
return None
metrics = analysis['metrics']
st.caption('Full-dimensional scores for the last shown relationship. No projection is used for these calculations.')
if 'direction_cosine' in metrics:
left, right = st.columns(2)
left.metric('Offset cosine', f'{metrics["direction_cosine"]:.4f}' if metrics['direction_cosine'] is not None else 'Undefined')
right.metric('Angle', f'{metrics["angle_degrees"]:.1f}°' if metrics['angle_degrees'] is not None else 'Undefined')
left.metric('Length ratio D−C / offset', f'{metrics["length_ratio"]:.4f}' if metrics['length_ratio'] is not None else 'Undefined')
right.metric('Prediction error (L2)', f'{metrics["prediction_error"]:.5f}')
else:
st.metric('Mean offset length', f'{metrics["offset_length"]:.5f}')
if len(analysis['pairs']) > 1:
agreement = metrics['example_agreement']
st.caption(f'Example offsets vs average: mean cosine {agreement:.4f}' if agreement is not None else 'Example direction agreement is undefined.')
if metrics['offset_length'] == 0 or metrics.get('observed_length') == 0:
st.info('A zero-length relationship has no direction. Direction cosine and angle are undefined for it.')
if st.button('Find analogy answers', key='find_analogy_answers', use_container_width=True):
st.session_state[f'neighbour_mode:{scope}'] = 'Analogy answers'
st.session_state[f'run_analogy:{scope}'] = True
if st.button('Add prediction as calc', key='save_relationship_prediction', disabled=len(vectors) >= 20, use_container_width=True):
save_calculation(metrics['prediction'], f'analogy prediction #{analysis["id"]}')
if st.button('Add mean offset as calc', key='save_relationship_offset', disabled=len(vectors) >= 20, use_container_width=True):
save_calculation(metrics['offset'], f'mean relationship offset #{analysis["id"]}')
st.caption('Prediction = C + mean(B − A). Analogy searches exclude the input vocabulary tokens. Scores use stored, unnormalised input vectors.')
return analysis
|