Toolkit documentation¶

chunk_length_samples: int = 30000

max_spikes_per_second: int = 16384

cd_iter: int = 0

coarse_cd: bool = True

threshold: float = 6.0

template_svd_compression_rank: int = 5

template_svd_compression_min_explained_variance: float = 0.005

up_factor: int = 4

upsampling_radius: int = 8

template_min_channel_amplitude: float = 1.0

amplitude_scaling_variance: float = 0.01 ** 2

amplitude_scaling_boundary: float = 1.0 / 3.0

max_iter: int = 100

channel_selection: Literal['template', 'amplitude'] = 'template'

channel_selection_radius: float | None = None

template_type: Literal['individual_compressed_upsampled', 'drifty', 'debug'] = 'drifty'

up_method: Literal['interpolation', 'keys3', 'keys4', 'direct'] = 'keys4'

drift_interp_params: InterpolationParams = tps_interp_clampna_extrap_params

upsampling_compression_map: Literal['yass', 'none'] = 'yass'

whitening: WhiteningConfig = WhiteningConfig(strategy='prewhiten_postapply')

whiten_features: bool = False

margin_factor: int = 2

peak_dt: int = 3

min_template_ptp: float = 1.0

always_keep_ptp: float = 10.0

min_template_snr: float = 0.0

min_template_count: int = 10

depth_order: bool = True

template_merge_cfg: TemplateMergeConfig | None = TemplateMergeConfig()

template_realignment_cfg: TemplateRealignmentConfig = TemplateRealignmentConfig()

precomputed_templates_npz: str | None = None

delete_pconv: bool = True

chunk_length_samples: int = 30000

fit_only: bool = False

detection_proposal: DetectionProposal = 'tpca'

proposal_filters: int = 1

propose_with_score_net: bool = False

voltage_threshold: float = 3.0

proposal_threshold: float = 5.0

score_net_threshold: float = 5.0

subtraction_threshold: float = 9.0

threshold_before_whitening: float = 10.0

peak_sign: PeakSign = 'both'

realign_to_denoiser: bool = True

denoiser_realignment_shift: int = 5

relative_peak_radius_samples: int = 5

relative_peak_radius_um: float | None = 35.0

spatial_dedup_radius_um: float | None = 50.0

temporal_dedup_radius_samples: int = 7

remove_exact_duplicates: bool = True

subtract_global_dedup: bool = True

positive_temporal_dedup_radius_samples: int = 41

subtract_radius_um: float = 200.0

trough_priority: float | None = 2.0

max_iter: int = 200

whiten: bool = True

denoise_before_localization: bool = False

denoise_before_amplitudes: bool = False

whiten_cfg: WhiteningConfig | None = WhiteningConfig(strategy='prewhiten_postapply')

subtraction_denoising_cfg: FeaturizationConfig = FeaturizationConfig(denoise_only=True, do_nn_denoise=True, tpca_denoise_fit='peeling', extract_radius=200.0, input_waveforms_name='raw', output_waveforms_name='subtracted')

first_denoiser_max_waveforms_fit: int = 512000

first_denoiser_noise_snips: int = 100 * 256

first_denoiser_noise_snip_length_mul: float = 2.5

first_denoiser_noise_density: float = 0.5

first_denoiser_thinning: float = 0.0

first_denoiser_temporal_jitter: int = 3

first_denoiser_spatial_jitter: float = 35.0

first_denoiser_spatial_dedup_radius: float = 100.0

save_iteration: bool = False

save_residnorm_decrease: bool = False

chunk_length_samples: int = 30000

detection_proposal: DetectionProposal = 'voltage'

proposal_filters: int = 1

voltage_threshold: float = 4.0

proposal_threshold: float = 5.0

max_spikes_per_chunk: int | None = None

peak_sign: Literal['pos', 'neg', 'both'] = 'both'

spatial_dedup_radius_um: float = 150.0

relative_peak_radius_um: float | None = 35.0

relative_peak_radius_samples: int = 5

temporal_dedup_radius_samples: int = 11

remove_exact_duplicates: bool = True

thinning: float = 0.0

time_jitter: int = 0

spatial_jitter_radius: float = 0.0

trough_priority: float | None = 2.0

shave_score: float = 10.0

skip: bool = False

extract_radius: float = 150.0

do_nn_denoise: bool = False

do_tpca_denoise: bool = True

tpca_denoise_fit: TPCADenoiserFitStage = 'peeling'

do_enforce_decrease: Literal['yes', 'no', 'loc_only'] = 'loc_only'

denoise_only: bool = False

save_input_voltages: bool = True

save_input_waveforms: bool = False

save_input_tpca_projs: bool = True

compute_input_tpca_projs_regardless: bool = False

save_output_waveforms: bool = False

save_output_tpca_projs: bool = False

save_collidedness: bool = False

save_amplitudes: bool = True

save_amplitude_vectors: bool = False

save_all_amplitudes: bool = False

output_amplitude_vectors: Literal['none', 'ptp', 'peak'] = 'none'

do_localization: bool = True

localization_radius: float = 100.0

localization_amplitude_type: Literal['peak', 'ptp'] = 'peak'

localization_decay_power: int = 1

localization_model: Literal['pointsource', 'dipole', 'gaussian'] = 'pointsource'

nn_localization: bool = True

additional_com_localization: bool = False

singlechan_denoised_amplitudes_and_localizations: bool = False

nn_denoiser_class_name: str = 'Decollider'

nn_denoiser_pretrained_path: str | None = None

nn_denoiser_train_epochs: int = 100

nn_denoiser_epoch_size: int = 200 * 256

nn_denoiser_extra_kwargs: dict | None = argfield(None, cli=False)

score_filter_radius_um: float | None = None

tpca_fit_radius: float = 75.0

tpca_rank: int = 8

tpca_centered: bool = False

learn_cleaned_tpca_basis: bool = False

vq_proposal_filters: int = 0

input_tpca_waveform_cfg: WaveformConfig | None = WaveformConfig(ms_before=0.75, ms_after=1.25)

tpca_max_waveforms: int = 40000

tpca_from_templates: bool = True

use_gmm_classifier: bool = False

pre_gmm_clustering_cfg: ClusteringConfig | None = None

pre_gmm_refinement_cfgs: Sequence[RefinementConfig | None] | None = None

gmm_refinement_cfg: RefinementConfig | None = None

gmm_clustering_features_cfg: ClusteringFeaturesConfig | None = None

fit_disabled_whitener: bool = False

whiten_cfg: WhiteningConfig | None = None

input_waveforms_name: str = 'collisioncleaned'

output_waveforms_name: str = 'denoised'

skip: bool = False

use_x: bool = True

use_z: bool = True

motion_aware: bool = True

use_amplitude: bool = False

use_signed_amplitude: bool = True

log_transform_amplitude: bool = False

amp_log_c: float = 5.0

amp_scale: float = 3.0

amplitude_kind: Literal['peak', 'ptp', 'rms', 'stored'] = 'ptp'

x_scale: float = 1.0

n_main_channel_pcs: int = 5

n_multi_channel_pcs: int = 0

multi_channel_pc_radius: float = 50.0

pc_scale: float = 2.0

pc_transform: Literal['log', 'sqrt', 'none'] | None = 'none'

pc_pre_transform_scale: float = 0.5

adaptive_feature_scales: bool = False

need_xyza: bool = False

feature_rank: int = 8

interp_params: InterpolationParams = tps_interp_params

motion_depth_mode: Literal['channel', 'localization'] = 'channel'

amplitudes_dataset_name: str = 'denoised_ptp_amplitudes'

voltages_dataset_name: str = 'collisioncleaned_voltages'

amplitude_vectors_dataset_name: str = 'denoised_ptp_amplitude_vectors'

localizations_dataset_name: str = 'point_source_localizations'

pca_dataset_name: str = 'collisioncleaned_tpca_features'

raise_for_numerics: bool = False

cluster_strategy: str = 'dpc'

sampling_cfg: FitSamplingConfig = default_clustering_fit_sampling_cfg

random_seed: int = 0

min_cluster_size: int = 5

knn_k: int | None = None

sigma_local: float = 5.0

sigma_regional: float | None = 25.0

n_neighbors_search: int = 50

radius_search: float = 25.0

noise_density: float = 0.0

outlier_radius: float | None = 25.0

outlier_neighbor_count: int = 10

dpc_kmeans_cleanup: bool = False

kmeanspp_initializations: int = 10

kmeans_iter: int = 100

components_per_channel: int = 20

component_overlap: float = 0.95

hellinger_strong: float = 0.0

hellinger_weak: float = 0.0

use_hellinger: bool = True

gmmdpc_max_sigma: float = 5.0

mop: bool = True

min_samples: int = 25

cluster_selection_epsilon: int = 1

recursive: bool = False

grid_dx: float = 15.0

grid_dz: float = 15.0

sklearn_class_name: str = 'DBSCAN'

sklearn_kwargs: dict | None = None

refinement_strategy: str = 'tmm'

sampling_cfg: FitSamplingConfig = default_clustering_fit_sampling_cfg

pc_merge_threshold: float = 0.1

pc_merge_metric: str = 'normeuc'

pc_merge_spikes_per_unit: int = 4096

pc_merge_linkage: str = 'complete'

pc_merge_rank: int = 5

pc_merge_min_iou: float = 0.95

cov_kind: str = 'factorizednoise'

glasso_alpha: float | int | None = None

neighb_overlap: float = 0.75

explore_neighb_steps: int = 0

min_count: int = 5

split_min_count: int = 5

channels_count_min: int = 1

signal_rank: int = 3

initialize_at_rank_0: bool = False

cl_alpha: float = 0.05

cl_split_only: bool = True

demolish_cl_alpha: float = 0.0

latent_prior_std: float = 1.0

initial_basis_shrinkage: float = 1.0

n_spikes_fit: int = 4096

distance_metric: ComponentDistanceMetric = 'scaled_normeuc'

n_candidates: int = 5

merge_group_size: int = 5

n_search: int | None = 3

n_explore: int | None = None

train_batch_size: int = 4096

eval_batch_size: int = 4096

split_friend_distance: float = 0.8

split_distance_threshold: float = 1.5

merge_distance_threshold: float = 1.5

criterion_em_iters: int = 3

hold_out_criterion: bool = True

n_em_iters: int = 250

em_converged_atol: float = 0.005

n_total_iters: int = 1

mixture_steps: Sequence[MixtureStep] = ('split', 'merge', 'demolish')

prior_pseudocount: float = 0.0

kmeansk: int = 4

single_split_k: int = 3

kmeans_tries: int = 10

kmeans_beta: float = 50.0

kmeanspp_tries: int = 5

kmeanspp_stop_rms: float = 5.0

kmeanspp_patience: int = 21

kmeanspp_greedy_proposals: int = 1

kmeanspp_neighb_overlap: float | None = None

kmeanspp_selection: KmeansppSelection = 'marginal'

kmeanspp_stopping: KmeansppStopping = 'dpmeanspp'

kmeanspp_sampling: KmeansppSampling = 'd2'

kmeanspp_alpha: float = 2.0

kmeanspp_z0: float = 3.0

full_proposal_every: int = 10

main_min_iters: int = 20

search_adj: Literal['top', 'explore'] = 'top'

robust_strategy: Literal['none', 'fixed'] = 'none'

robust_fixed_std_dataset: str = 'collidedness'

robust_fixed_power: float = 40.0

robust_df: float = 4.0

demolition_min_resp_ratio: float = 0.0

demolish_during_selection: bool = False

refit_in_demolition: bool = False

em_after_demolish: bool = True

whiten_split: bool = True

fix_responsibilities_in_split: bool = True

scale_dist_args: tuple[float, float, float] = (0.01, 3.0 / 4.0, 4.0 / 3.0)

whiten_dist: bool = True

template_merge_cfg: TemplateMergeConfig | None = TemplateMergeConfig(linkage='single')

glom_force_merge_template_distance: float = 0.3

glom_violation_ms: float = 1.0

glom_jitter_ms: float = 20.0

glom_min_violation_evidence: float = 10.0

glom_violation_threshold: float | None = 0.3

glom_veto_threshold: float | None = None