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