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

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
rich.kr...@modernizingmedicine.com wrote:
 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