[jvm-packages] separate classification and regression model and integrate with ML package (#1608)

This commit is contained in:
Nan Zhu
2016-09-30 11:49:03 -04:00
committed by GitHub
parent 3b9987ca9c
commit 1673bcbe7e
16 changed files with 771 additions and 381 deletions

View File

@@ -38,6 +38,8 @@ trait EvalTrait extends IEvaluation {
def eval(predicts: Array[Array[Float]], dmat: DMatrix): Float
private[scala] def eval(predicts: Array[Array[Float]], jdmat: java.DMatrix): Float = {
require(predicts.length == jdmat.getLabel.length, "predicts size and label size must match " +
s" predicts size: ${predicts.length}, label size: ${jdmat.getLabel.length}")
eval(predicts, new DMatrix(jdmat))
}
}