diff --git a/h2o-core/src/main/java/hex/ConfusionMatrix.java b/h2o-core/src/main/java/hex/ConfusionMatrix.java index eefd739c4a4b..0cfb23f4e4d9 100644 --- a/h2o-core/src/main/java/hex/ConfusionMatrix.java +++ b/h2o-core/src/main/java/hex/ConfusionMatrix.java @@ -86,7 +86,7 @@ public double err() { double err = n; for( int d = 0; d < _cm.length; ++d ) err -= _cm[d][d]; - return (double) err / n; + return err / n; } public double errCount() { double err = totalRows(); @@ -224,7 +224,7 @@ private TwoDimTable toTable() { for( int p=0; p< _cm[a].length; p++ ) { sum += _cm[a][p]; preds[p] += _cm[a][p]; - isInt &= (_cm[a][p] == (int)_cm[a][p]); + isInt &= (_cm[a][p] == (long)_cm[a][p]); } acts[a] = sum; } @@ -242,7 +242,7 @@ private TwoDimTable toTable() { String[] colType = new String[colHeader.length]; String[] colFormat = new String[colHeader.length]; for (int i=0; i