sgoldenlab / simba

SimBA (Simple Behavioral Analysis), a pipeline and GUI for developing supervised behavioral classifiers
https://simba-uw-tf-dev.readthedocs.io/
GNU General Public License v3.0
289 stars 141 forks source link

Model Training Error #235

Closed kylethieringer closed 1 year ago

kylethieringer commented 1 year ago

Describe the bug When trying to train new model I am running into an error in the random under sampling function.

To Reproduce Steps to reproduce the behavior:

  1. load project config
  2. train single model

Screenshots

Screenshot 2023-03-02 at 1 28 01 AM

Desktop (please complete the following information):

Additional context Im not sure if this is important but I uploaded tracks from sleap h5 files and labels from deepethogram. I have 2 animals with 13pts each. I am trying to classify a sparse behavior <10% of frames.

sronilsson commented 1 year ago

Thank you @kylethieringer! I inserted the bout sampler train/test option recently, haven't had any feedback yet, so might be a few bugs to clean up, so this is helpful.

The error came from the bout sampling function returning arrays, and the downstream random undersampler accepting dataframes. If both bout sampling and the random undersampler was used, you'd bump into that error.

I fixed it in version 1.51.4 - could you update with pip install simba-uw-tf-dev --upgrade and see if it works on your end?

kylethieringer commented 1 year ago

Thanks so much for your speedy response and fix! No issues anymore!