| """Show random samples. Simple method, but it often turns up interesting things.""" |
| import pandas as pd |
| import streamlit as st |
|
|
| from src.subpages.page import Context, Page |
| from src.utils import htmlify_labeled_example |
|
|
|
|
| class RandomSamplesPage(Page): |
| name = "Random Samples" |
| icon = "shuffle" |
|
|
| def _get_widget_defaults(self): |
| return { |
| "random_sample_size_min": 128, |
| } |
|
|
| def render(self, context: Context): |
| st.title("🎲 Random Samples") |
| with st.expander("💡", expanded=True): |
| st.write( |
| "Show random samples. Simple method, but it often turns up interesting things." |
| ) |
|
|
| random_sample_size = st.number_input( |
| "Random sample size:", |
| value=min(st.session_state.random_sample_size_min, context.split_sample_size), |
| step=16, |
| key="random_sample_size", |
| ) |
|
|
| if st.button("🎲 Resample"): |
| st.experimental_rerun() |
|
|
| random_indices = context.df.sample(int(random_sample_size)).index |
| samples = context.df_tokens_merged.loc[random_indices] |
|
|
| for i, idx in enumerate(random_indices): |
| sample = samples.loc[idx] |
|
|
| if isinstance(sample, pd.Series): |
| continue |
|
|
| col1, _, col2 = st.columns([0.08, 0.025, 0.8]) |
|
|
| counter = f"<span title='#sample | index' style='display: block; background-color: black; opacity: 1; color: wh^; padding: 0 5px'>[{i+1} | {idx}]</span>" |
| loss = f"<span title='total loss' style='display: block; background-color: yellow; color: gray; padding: 0 5px;'>𝐿 {sample.losses.sum():.3f}</span>" |
| col1.write(f"{counter}{loss}", unsafe_allow_html=True) |
| col1.write("") |
| col2.write(htmlify_labeled_example(sample), unsafe_allow_html=True) |
|
|
