yaohungt / Multimodal-Transformer

[ACL'19] [PyTorch] Multimodal Transformer
MIT License
827 stars 152 forks source link

How can I set the weights for the imbalanced classification? #35

Open curiousRed opened 3 years ago

curiousRed commented 3 years ago

I am dealing with the imbalanced classed of the text + tabular data model. Is there a way to set up the class weights? Thank you very much!