google / automl

Google Brain AutoML
Apache License 2.0
6.22k stars 1.45k forks source link

How to add class weights? #1205

Open alexw92 opened 10 months ago

alexw92 commented 10 months ago

I am currently using efficient det lite family to train on a custom dataset but the data is imbalanced. I alsready worked with an "other" class but the results are still not fine. I would like to introduce class weights during training but I dont know where that is done? Has anyone ever done this or is it somehow natively supported through some param I could not find so far?