Waikato / wekaDeeplearning4j

Weka package for the Deeplearning4j java library
https://deeplearning.cms.waikato.ac.nz/
GNU General Public License v3.0
185 stars 202 forks source link

Plot Top Losses #69

Open basedrhys opened 3 years ago

basedrhys commented 3 years ago

Is your feature request related to a problem? Please describe. One helpful model debugging tool is to plot the instances which our model gave highest loss for -- where our predictions were most different to ground truth. This provides some useful feedback on edge cases / areas our data may be faulty.

Describe the solution you'd like A similar functionality in WDL4J, initially for image models but potentially all models trained in WEKA

Additional context Notebook

example