Asher Krim Krimit
Krimit
/ gist:f61b5e3be2b1e380d1ca
Created
July 6, 2015 20:57
return scores in multiclass org.apache.spark.mllib.classification.LogisticRegressionModel
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| override protected def predictPoint( | |
| dataMatrix: Vector, | |
| weightMatrix: Vector, | |
| intercept: Double) = { | |
| require(dataMatrix.size == numFeatures) | |
| var (margins, maxMargin, bestClass) = computeMargins(dataMatrix, weightMatrix, intercept) | |
| if (numClasses == 2) { | |
| val scores = computeScores(margins) |