[SPARK-20199] GradientBoostedTreesModel doesn't have featureSubsetStrategy parameter

• Type: Improvement
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 2.1.0
• Fix Version/s: 2.3.0
• Component/s: ML, MLlib
• Assignee: Pralabh Kumar
• Reporter: Pralabh Kumar
• Created: 03/Apr/17 06:55
• Updated: 10/Nov/17 11:20
• Resolved: 10/Nov/17 11:17

Description

Spark GradientBoostedTreesModel doesn't have featureSubsetStrategy . It Uses random forest internally ,which have featureSubsetStrategy hardcoded "all". It should be provided by the user to have randomness at the feature level.

This parameter is available in H2O and XGBoost.

Sample from H2O.ai
gbmParams._col_sample_rate

Please provide the parameter .

Issue Links

• Is contained by: SPARK-14047 GBT improvement umbrella
• links to: [Github] Pull Request #18118 (pralabhkumar)