qanastek / HugsVision

HugsVision is a easy to use huggingface wrapper for state-of-the-art computer vision

Home Page:https://pypi.org/project/hugsvision/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add confusion matrix for VisionClassifierTrainer

qanastek opened this issue · comments

Return: from sklearn.metrics import confusion_matrix
In the prection example: from sklearn.metrics import ConfusionMatrixDisplay