documented confusion-matrix

scicloj · Mar 21, 2024 · 614f900 · 614f900
1 parent f82bc40
commit 614f900
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/src/scicloj/metamorph/ml/viz.clj b/src/scicloj/metamorph/ml/viz.clj
@@ -54,11 +54,18 @@
 
 
 (defn confusion-matrix
-  (
-   [predicted-labels labels
+  "Generates a confusin matrix plot out of `predicted-labels` and `labels`
+
+   `opts`
+     `normalize` : Can be :none (default) or :all and decides if the values in the matrix are counts or percentages.
+
+   `hanami-opts` Options passed to hanami to control the plot. Can be any of the default hanami
+   substituions keys.
+
+  "
+  ([predicted-labels labels
     opts
     hanami-opts]
-
    (->
     (scicloj.metamorph.ml.viz.confusionmatrix/cm-values predicted-labels labels opts)
     (scicloj.metamorph.ml.viz.confusionmatrix/confusion-matrix-chart)

diff --git a/src/scicloj/metamorph/ml/viz/confusionmatrix.clj b/src/scicloj/metamorph/ml/viz/confusionmatrix.clj
@@ -1,14 +1,12 @@
 (ns scicloj.metamorph.ml.viz.confusionmatrix
   (:require
-   [aerial.hanami.common :as hc]
    [aerial.hanami.templates :as ht]
    [scicloj.metamorph.ml.classification :as cl]
    [scicloj.metamorph.ml.learning-curve]
    [scicloj.metamorph.ml.viz.learning-curve]))
 
 
 
-
 (def layer
   [(assoc ht/rect-layer :encoding ht/xy-encoding)
    (assoc ht/text-layer