[SPARK-32298] tree models prediction optimization

• Type: Improvement
• Status: Resolved
• Priority: Minor
• Resolution: Fixed
• Affects Version/s: 3.1.0
• Fix Version/s: 3.1.0
• Component/s: ML
• Assignee: Ruifeng Zheng
• Reporter: Ruifeng Zheng
• Created: 14/Jul/20 04:53
• Updated: 17/Jul/20 17:01
• Resolved: 17/Jul/20 17:01

Description

in Node's method

 

def predictImpl(features: Vector): LeafNode

 

use while-loop instead of the recursive way.

Issue Links

• links to: [Github] Pull Request #29095 (zhengruifeng)