correct metrics with ignore class

ELEKTRONN / elektronn3

A PyTorch-based library for working with 3D and 2D convolutional neural networks, with focus on semantic segmentation of volumetric biomedical image data

MIT License

160 stars 27 forks source link

correct metrics with ignore class #31

Closed Optiligence closed 5 years ago

Optiligence commented 5 years ago

just as inspiration this is obviously not ok as default the ignore class id(s) should somehow end up in that function but e3 currently doesn’t know about them (the class_weights) at all

mdraw commented 5 years ago

Thanks, this functionality is now implemented via ea516afe9ff (now configurable and optional).