Package

org.apache.spark.mllib.tree

configuration

Permalink

package configuration

Visibility

Public
All

Type Members

case class BoostingStrategy(treeStrategy: Strategy, loss: Loss, numIterations: Int = 100, learningRate: Double = 0.1, validationTol: Double = 1e-5) extends Serializable with Product

:: Experimental :: Configuration options for org.apache.spark.mllib.tree.GradientBoostedTrees.
:: Experimental :: Configuration options for org.apache.spark.mllib.tree.GradientBoostedTrees.
treeStrategy
Parameters for the tree algorithm. We support regression and binary classification for boosting. Impurity setting will be ignored.
loss
Loss function used for minimization during gradient boosting.
numIterations
Number of iterations of boosting. In other words, the number of weak hypotheses used in the final model.
learningRate
Learning rate for shrinking the contribution of each estimator. The learning rate should be between in the interval (0, 1]
validationTol
Useful when runWithValidation is used. If the error rate on the validation input between two iterations is less than the validationTol then stop. Ignored when org.apache.spark.mllib.tree.GradientBoostedTrees.run() is used.

Annotations
@Since( "1.2.0" ) @Experimental()
class Strategy extends Serializable

:: Experimental :: Stores all the configuration options for tree construction
:: Experimental :: Stores all the configuration options for tree construction

Annotations
@Since( "1.0.0" ) @Experimental()

Value Members

object Algo extends Enumeration

:: Experimental :: Enum to select the algorithm for the decision tree
:: Experimental :: Enum to select the algorithm for the decision tree

Annotations
@Since( "1.0.0" ) @Experimental()
object BoostingStrategy extends Serializable

Annotations
@Since( "1.2.0" ) @Experimental()
object FeatureType extends Enumeration

:: Experimental :: Enum to describe whether a feature is "continuous" or "categorical"
:: Experimental :: Enum to describe whether a feature is "continuous" or "categorical"

Annotations
@Since( "1.0.0" ) @Experimental()
object QuantileStrategy extends Enumeration

:: Experimental :: Enum for selecting the quantile calculation strategy
:: Experimental :: Enum for selecting the quantile calculation strategy

Annotations
@Since( "1.0.0" ) @Experimental()
object Strategy extends Serializable

Annotations
@Since( "1.2.0" ) @Experimental()

Ungrouped