search

beacon-biosignals / OndaBatches.jl

Local and distributed batch loading for Onda datasets
MIT License
2 stars 0 forks source link

Allow choosing integer type in `OndaBatches.int_encode_labels` #40

Open ericphanson opened 7 months ago

ericphanson commented 7 months ago

This is a pretty useful function (we are using it elsewhere), but it would be nice if the user could specify the type by means of the encoding dictionary. E.g. yasa uses 

Dict("no_stage" => -2,
                     "wake" => 0,
                     "nrem1" => 1,
                     "nrem2" => 2,
                     "nrem3" => 3,
                     "rem" => 4)

which needs signed ints.