yogsototh/clj-ml
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Don't print evaluation results, just collect them.

Browse source
This commit is contained in:
Joshua Eckroth 2013-03-25 16:27:33 -04:00
parent d49820eda5
commit aab3d42a99
1 changed files with 29 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/clj_ml/classifiers.clj
View file

@ -492,11 +492,9 @@
(defn- collect-evaluation-results
"Collects all the statistics from the evaluation of a classifier."
([class-labels ^Evaluation evaluation]
(do
(println (.toMatrixString evaluation))
(println "=== Summary ===")
(println (.toSummaryString evaluation))
{:correct (try-metric #(.correct evaluation))
{:confusion-matrix (.toMatrixString evaluation)
:summary (.toSummaryString evaluation)
:correct (try-metric #(.correct evaluation))
:incorrect (try-metric #(.incorrect evaluation))
:unclassified (try-metric #(.unclassified evaluation))
:percentage-correct (try-metric #(.pctCorrect evaluation))
@ -521,7 +519,7 @@
:f-measure (try-multiple-values-metric class-labels (fn [i] (try-metric #(.fMeasure evaluation i))))
:precision (try-multiple-values-metric class-labels (fn [i] (try-metric #(.precision evaluation i))))
:recall (try-multiple-values-metric class-labels (fn [i] (try-metric #(.recall evaluation i))))
:evaluation-object evaluation})))
:evaluation-object evaluation}))
(defmulti classifier-evaluate
"Evaluates a trained classifier using the provided dataset or cross-validation.