bacpipe.model_pipelines.feature_extractors.bat

Functions

getSequences(spectrogram, patch_len, ...)

prepareData(y)

slideWindow(a, size, step)

Classes

BAT(max_len, patch_dim, d_model, num_classes)

Classifier based on a pytorch TransformerEncoder.

Model([threshold])

ModelBaseClass(sr, segment_length, model_name)

class bacpipe.model_pipelines.feature_extractors.bat.Model(threshold=0.5, **kwargs)[source]

Bases: ModelBaseClass

classifier_predictions(cls_token)[source]
preprocess(audio)[source]