NifTK / NiftyNet

[unmaintained] An open-source convolutional neural networks platform for research in medical image analysis and image-guided therapy
http://niftynet.io
Apache License 2.0
1.36k stars 403 forks source link

weighted cross entropy for classification task #408

Open koriavinash1 opened 5 years ago

koriavinash1 commented 5 years ago

Is it possible to add class specific weights in cross entropy loss? what variables in config files correspond to these weights?