water.test.util

Class ConfusionMatrixUtils

java.lang.Object

water.test.util.ConfusionMatrixUtils

public class ConfusionMatrixUtils
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
ConfusionMatrixUtils()

Method Summary

Modifier and Type	Method and Description
static void	assertCMEqual(java.lang.String[] expectedDomain, double[][] expectedCM, hex.ConfusionMatrix actualCM)
static hex.ConfusionMatrix	buildCM(water.fvec.Vec actuals, water.fvec.Vec predictions) Build the CM data from the actuals and predictions, using the default threshold.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ConfusionMatrixUtils

public ConfusionMatrixUtils()

Method Detail

buildCM

public static hex.ConfusionMatrix buildCM(water.fvec.Vec actuals,
                                          water.fvec.Vec predictions)

Build the CM data from the actuals and predictions, using the default threshold. Print to Log.info if the number of classes is below the print_threshold. Actuals might have extra levels not trained on (hence never predicted). Actuals with NAs are not scored, and their predictions ignored.

assertCMEqual

public static void assertCMEqual(java.lang.String[] expectedDomain,
                                 double[][] expectedCM,
                                 hex.ConfusionMatrix actualCM)