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
287 stars 140 forks source link

Sklearn visualizations of single videos #89

Closed AyusoAa closed 3 years ago

AyusoAa commented 3 years ago

Dear SimBA developers, I'd like to generate one single exemplary video with sklearn visualization of model predictions, instead I SimBA generates visualizations of all videos at once, if I am right. Would it be possible to add such feature, similar to the ROI visualizations? Best, Irene

sgoldenlab commented 3 years ago

Hello @AyusoAa ! Yes we are on it and I will let you know when it's done - if you cannot wait: SimBA grabs all of the videos directly in the project_folder/csv/machine_results and plots the results when you click the sklearn visualization. If you want to do just a select few, put the videos you do not want the visualize in a temporarily folder somewhere else, and then move them back.

sgoldenlab commented 3 years ago

@AyusoAa, I have implemented what you have suggested. Please update to simba-uw-tf==1.3.3 and let us know if that works for you.

sronilsson commented 3 years ago

Note you should see it in the dev version too - image

sgoldenlab commented 3 years ago

@AyusoAa - did this update work as expected?

AyusoAa commented 3 years ago

It does work! Thank you so much for your attention and work :)