spark-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From Pralabh Kumar <>
Subject featureSubsetStrategy parameter for GradientBoostedTreesModel
Date Fri, 16 Jun 2017 02:23:06 GMT
Hi everyone

Currently GBT doesn't expose featureSubsetStrategy as exposed by Random

GradientBoostedTrees in Spark have hardcoded feature subset strategy to
"all" while calling random forest in  DecisionTreeRegressor.scala

val trees =, oldStrategy, numTrees = 1,
featureSubsetStrategy = "all",

It should provide functionality to the user to set featureSubsetStrategy
("auto", "all" ,"sqrt" , "log2" , "onethird") ,
the way random forest does.

This will help GBT to have randomness at feature level.

Jira SPARK-20199 <>

Please let me know , if my understanding is correct.

Pralabh Kumar

View raw message