search

hendrycks / emodiversity

Wellbeing and Emotion Prediction (NeurIPS 2022)
MIT License
8 stars 0 forks source link

Change VCE dataloader to return vector instead of max label #7

Open JunShern opened 2 years ago

JunShern commented 2 years ago

Instead of only returning the max class for each video, simply return the full vector of scores (vector of length 27), eg.

Label: [0.06923076923076923, 0.0, 0.0, 0.15384615384615385, 0.0, 0.0, 0.0, 0.14615384615384613, 0.2076923076923077, 0.0, 0.0, 0.0, 0.0, 0.16153846153846155, 0.0, 0.038461538461538464, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.07692307692307693, 0.0, 0.0, 0.11538461538461538]