Add colors to confusion matrix and classification report

2024-05-14 00:41:29 +02:00
parent 8d20545fd2
commit 99c9c6731f
5 changed files with 15 additions and 11 deletions
--- a/src/main/Scores.h
+++ b/src/main/Scores.h
@@ -17,7 +17,7 @@ namespace platform {
        float precision(int num_class);
        float recall(int num_class);
        torch::Tensor get_confusion_matrix() { return confusion_matrix; }
-        std::string classification_report();
+        std::string classification_report(std::string color = "");
        json get_confusion_matrix_json(bool labels_as_keys = false);
        void aggregate(const Scores& a);
    private: