examples icon indicating copy to clipboard operation
examples copied to clipboard

added class_weight for imbalanced datasets

Open pdubz-sudo opened this issue 3 years ago • 3 comments

  • added class_weight to create and train method in image_classifier.py
  • added class_weight to train_model method in train_image_classifier.py All this was done so that the class_weight can be added to the model.fit() because Keras Models already have this as an argument.

I was not able to actually test if this works because I can't build from source when I tried from the readme instructions for model maker (I'm not sure why I can't build). So, these are the changes I think will work. Keras Model already has class_weight as an argument in the .fit() so I just tried to change the code so that a class_weight can be added from the .create() of the model_maker for image classification.

pdubz-sudo avatar Apr 23 '21 09:04 pdubz-sudo

Thanks for your pull request. It looks like this may be your first contribution to a Google open source project (if not, look below for help). Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

:memo: Please visit https://cla.developers.google.com/ to sign.

Once you've signed (or fixed any issues), please reply here with @googlebot I signed it! and we'll verify it.


What to do if you already signed the CLA

Individual signers
Corporate signers

ℹ️ Googlers: Go here for more info.

google-cla[bot] avatar Apr 23 '21 09:04 google-cla[bot]

I made this feature request and am hoping this PR solves it https://github.com/tensorflow/tensorflow/issues/48703

pdubz-sudo avatar Apr 23 '21 10:04 pdubz-sudo

@googlebot I signed it!

pdubz-sudo avatar Apr 23 '21 10:04 pdubz-sudo