* [Feature] Support calculate confusion matrix and plot it. * Fix keepdim * Update confusion_matrix tools and the plot graph. * Revert accidental modification. * Update docstring * Move confusion matrix tool to |
||
---|---|---|
.. | ||
analysis_tools | ||
misc | ||
model_converters | ||
torchserve | ||
visualizations | ||
dist_test.sh | ||
dist_train.sh | ||
kfold-cross-valid.py | ||
slurm_test.sh | ||
slurm_train.sh | ||
test.py | ||
train.py |