Re: Top N predictions

2014-07-11 Thread Sean Owen
I don't believe it is. Recently when I needed to do this, I just copied out the underlying probability / margin function and calculated it from the model params. It's just a dot product. On Fri, Jul 11, 2014 at 7:48 PM, Rich Kroll wrote: > Hello all, > In our use case we would like to return top

Top N predictions

2014-07-11 Thread Rich Kroll
Hello all, In our use case we would like to return top 10 predicted values. I've looked at NaiveBayes & LogisticRegressionModel and cannot seem to find a way to get the predicted values for a vector - is this possible with mllib/spark? Thanks, Rich