search

OlafenwaMoses / ImageAI

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
https://www.genxr.co/#products
MIT License
8.58k stars 2.19k forks source link

Generating confusion matrix for custom training #541

Open r-arvind opened 4 years ago

r-arvind commented 4 years ago

Confusion matrix is an essential metric for classification problem. As of now, imageAi does not generate confusion matrix while training/testing on custom dataset

AbimbolaOO commented 4 years ago

Thank you for pointing this out. I think this is a really could feature that the future version of imageAI might incorporate.