pvtend.ClassifyResult

class pvtend.ClassifyResult(stage_all: dict[str, set[int]], stage_awb: dict[str, set[int]], stage_cwb: dict[str, set[int]], stage_neu: dict[str, set[int]], h_scale: float | None, stages: list[str], classify_levels: list[int], classify_threshold: int)

Holds RWB variant classification results.

stage_all

{stage: set_of_track_ids}

Type:

dict[str, set[int]]

stage_awb

{stage: set_of_AWB_track_ids}

Type:

dict[str, set[int]]

stage_cwb

{stage: set_of_CWB_track_ids}

Type:

dict[str, set[int]]

stage_neu

{stage: set_of_NEUTRAL_track_ids}

Type:

dict[str, set[int]]

h_scale

Captured from the first NPZ file.

Type:

float | None

stages

Ordered stage names.

Type:

list[str]

classify_levels

Pressure levels used.

Type:

list[int]

classify_threshold

Threshold used.

Type:

int

__init__(stage_all: dict[str, set[int]], stage_awb: dict[str, set[int]], stage_cwb: dict[str, set[int]], stage_neu: dict[str, set[int]], h_scale: float | None, stages: list[str], classify_levels: list[int], classify_threshold: int) → None

Methods

__init__(stage_all, stage_awb, stage_cwb, ...)
load(path)	Load from a previously-saved PKL.
save(path)	Persist to pickle (same format as core script).

Attributes

stage_labels	{stage: {track_id: label}} where label ∈ AWB/CWB/NEUTRAL/Omega.
stage_tracksets
variant_trackset	Variant → frozenset mapping, e.g. AWB_onset.
stage_all
stage_awb
stage_cwb
stage_neu
h_scale
stages
classify_levels
classify_threshold