org.apache.spark.mllib.tree.configuration

Class Strategy

Object

org.apache.spark.mllib.tree.configuration.Strategy

All Implemented Interfaces:

java.io.Serializable

public class Strategy
extends Object
implements scala.Serializable

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

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
Strategy(scala.Enumeration.Value algo, Impurity impurity, int maxDepth, int maxBins, scala.Enumeration.Value quantileCalculationStrategy, scala.collection.immutable.Map<Object,Object> categoricalFeaturesInfo, int maxMemoryInMB)

Method Summary

Methods

Modifier and Type	Method and Description
scala.Enumeration.Value	algo()
scala.collection.immutable.Map<Object,Object>	categoricalFeaturesInfo()
Impurity	impurity()
int	maxBins()
int	maxDepth()
int	maxMemoryInMB()
scala.Enumeration.Value	quantileCalculationStrategy()

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Strategy

public Strategy(scala.Enumeration.Value algo,
        Impurity impurity,
        int maxDepth,
        int maxBins,
        scala.Enumeration.Value quantileCalculationStrategy,
        scala.collection.immutable.Map<Object,Object> categoricalFeaturesInfo,
        int maxMemoryInMB)

Method Detail

algo

public scala.Enumeration.Value algo()

impurity

public Impurity impurity()

maxDepth

public int maxDepth()

maxBins

public int maxBins()

quantileCalculationStrategy

public scala.Enumeration.Value quantileCalculationStrategy()

categoricalFeaturesInfo

public scala.collection.immutable.Map<Object,Object> categoricalFeaturesInfo()

maxMemoryInMB

public int maxMemoryInMB()