[SPARK-30358] ML expose predictRaw and predictProbability

• Type: Improvement
• Status: Resolved
• Priority: Minor
• Resolution: Fixed
• Affects Version/s: 3.0.0
• Fix Version/s: 3.0.0
• Component/s: ML
• Assignee: Ruifeng Zheng
• Reporter: Ruifeng Zheng
• Created: 26/Dec/19 11:18
• Updated: 03/Jan/20 18:47
• Resolved: 31/Dec/19 04:50

Description

Support predictRaw and predictProbability on single instance.

Currently we have expose predict method for classification/regression/clustering models. (GMM has already exposed predictProbability method.)

We may also need to expose predictRaw and predictProbability for online prediction

 

Issue Links

• relates to: SPARK-25097 Support prediction on single instance in KMeans/BiKMeans/GMM
• relates to: SPARK-10413 ML models should support prediction on single instances
• links to: GitHub Pull Request #27015
• links to: GitHub Pull Request #27082