Your team needs to build a model that predicts whether images contain a driver's license, passport, or credit card. The data engineering team already built the pipeline and generated a dataset composed of 10,000 images with driver's licenses, 1,000 images with passports, and 1,000 images with credit cards. You now have to train a model with the following label map: ['driversjicense', 'passport', 'credit_card']. Which loss function should you use?
se sparse_categorical_crossentropy. Examples for above 3-class classification problem: [1] , [2], [3]
Currently there are no comments in this discussion, be the first to comment!