quietset 0.16.0

Filter datasets by label stability across evaluators, budgets, seeds, and models
Documentation 
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

use crate::observation::Observation;
use indexmap::IndexMap;

/// Groups observations by sample_id, preserving insertion order.
pub fn group_by_sample_id(
    observations: impl Iterator<Item = Observation>,
) -> IndexMap<String, Vec<Observation>> {
    let mut groups: IndexMap<String, Vec<Observation>> = IndexMap::new();
    for obs in observations {
        groups.entry(obs.sample_id.clone()).or_default().push(obs);
    }
    groups
}

/// Groups observations by `block_id`, preserving insertion order. Observations with no
/// `block_id` are skipped — they carry no block-level stability information.
///
/// `layer_id` is deliberately not part of this key. A training block spans multiple layers,
/// and `trajectory_audit`'s per-layer `dead_unit_delta`/`saturated_unit_delta` report those
/// layers *within* one block's observations. Keying on `(layer_id, block_id)` instead would
/// split one legitimate block into unrelated per-layer groups.
pub fn group_by_block_id(
    observations: impl Iterator<Item = Observation>,
) -> IndexMap<String, Vec<Observation>> {
    let mut groups: IndexMap<String, Vec<Observation>> = IndexMap::new();
    for obs in observations {
        if let Some(block_id) = obs.block_id.clone() {
            groups.entry(block_id).or_default().push(obs);
        }
    }
    groups
}