API Reference

API Reference#

Submodules#

autosort_neuron.auto_sorting module#

autosort_neuron.config module#

autosort_neuron.detection module#

autosort_neuron.model module#

class autosort_neuron.model.AE(*args, **kwargs)[source]#

Bases: Module

forward(features)[source]#

class autosort_neuron.model.AutoSort(ch_num, samplepoints, loc_dim, device, set_shank_id, save_dir, pos_weight_noise=None, pos_weight_label=None)[source]#

Bases: object

eval()[source]#

iter_model(batch_features, classify_labels, labels, single_waveform, pred_loc)[source]#

iter_model_eval(batch_features, classify_labels, labels, single_waveform, pred_loc)[source]#

iter_model_eval_umap(batch_features, classify_labels, labels, single_waveform, pred_loc)[source]#

load_model()[source]#

save_model()[source]#

to_device(device)[source]#

train()[source]#

class autosort_neuron.model.clssimp(*args, **kwargs)[source]#

Bases: Module

forward(x)[source]#

intermediate_forward(x)[source]#

autosort_neuron.model.load_day_length(path)[source]#

autosort_neuron.run module#

autosort_neuron.sma module#

autosort_neuron.sorting module#

autosort_neuron.utils module#

autosort_neuron.utils.display_top(snapshot, key_type='lineno', limit=3)[source]#

autosort_neuron.utils.seed_all(seed_value, cuda_deterministic=False)[source]#

autosort_neuron.waveform_loader module#

autosort_neuron.waveform_loader.location_cal(sensor_positions, batch_features)[source]#

autosort_neuron.waveform_loader.location_cal_group(sensor_positions, batch_features, group_id)[source]#

class autosort_neuron.waveform_loader.waveformLoader(*args, **kwargs)[source]#: Bases: Dataset