Strategy (Spark 1.2.1 JavaDoc)

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 numClasses, int maxBins, scala.Enumeration.Value quantileCalculationStrategy, scala.collection.immutable.Map<Object,Object> categoricalFeaturesInfo, int minInstancesPerNode, double minInfoGain, int maxMemoryInMB, double subsamplingRate, boolean useNodeIdCache, scala.Option<String> checkpointDir, int checkpointInterval)`
`Strategy(scala.Enumeration.Value algo, Impurity impurity, int maxDepth, int numClasses, int maxBins, java.util.Map<Integer,Integer> categoricalFeaturesInfo)` Java-friendly constructor for `Strategy`

Method Summary

Methods
Modifier and Type	Method and Description
`scala.Enumeration.Value`	`algo()`
`void`	`assertValid()` Check validity of parameters.
`scala.collection.immutable.Map<Object,Object>`	`categoricalFeaturesInfo()`
`scala.Option<String>`	`checkpointDir()`
`int`	`checkpointInterval()`
`Strategy`	`copy()` Returns a shallow copy of this instance.
`static Strategy`	`defaultStrategy(String algo)` Construct a default set of parameters for `DecisionTree`
`scala.Enumeration.Value`	`getAlgo()`
`scala.collection.immutable.Map<Object,Object>`	`getCategoricalFeaturesInfo()`
`scala.Option<String>`	`getCheckpointDir()`
`int`	`getCheckpointInterval()`
`Impurity`	`getImpurity()`
`int`	`getMaxBins()`
`int`	`getMaxDepth()`
`int`	`getMaxMemoryInMB()`
`double`	`getMinInfoGain()`
`int`	`getMinInstancesPerNode()`
`int`	`getNumClasses()`
`scala.Enumeration.Value`	`getQuantileCalculationStrategy()`
`double`	`getSubsamplingRate()`
`boolean`	`getUseNodeIdCache()`
`Impurity`	`impurity()`
`boolean`	`isMulticlassClassification()`
`boolean`	`isMulticlassWithCategoricalFeatures()`
`int`	`maxBins()`
`int`	`maxDepth()`
`int`	`maxMemoryInMB()`
`double`	`minInfoGain()`
`int`	`minInstancesPerNode()`
`int`	`numClasses()`
`scala.Enumeration.Value`	`quantileCalculationStrategy()`
`void`	`setAlgo(String algo)` Sets Algorithm using a String.
`void`	`setCategoricalFeaturesInfo(java.util.Map<Integer,Integer> categoricalFeaturesInfo)` Sets categoricalFeaturesInfo using a Java Map.
`void`	`setCheckpointDir(scala.Option<String> x$1)`
`void`	`setCheckpointInterval(int x$1)`
`void`	`setImpurity(Impurity x$1)`
`void`	`setMaxBins(int x$1)`
`void`	`setMaxDepth(int x$1)`
`void`	`setMaxMemoryInMB(int x$1)`
`void`	`setMinInfoGain(double x$1)`
`void`	`setMinInstancesPerNode(int x$1)`
`void`	`setNumClasses(int x$1)`
`void`	`setQuantileCalculationStrategy(scala.Enumeration.Value x$1)`
`void`	`setSubsamplingRate(double x$1)`
`void`	`setUseNodeIdCache(boolean x$1)`
`double`	`subsamplingRate()`
`boolean`	`useNodeIdCache()`

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 numClasses,
        int maxBins,
        scala.Enumeration.Value quantileCalculationStrategy,
        scala.collection.immutable.Map<Object,Object> categoricalFeaturesInfo,
        int minInstancesPerNode,
        double minInfoGain,
        int maxMemoryInMB,
        double subsamplingRate,
        boolean useNodeIdCache,
        scala.Option<String> checkpointDir,
        int checkpointInterval)

Strategy

public Strategy(scala.Enumeration.Value algo,
        Impurity impurity,
        int maxDepth,
        int numClasses,
        int maxBins,
        java.util.Map<Integer,Integer> categoricalFeaturesInfo)

Java-friendly constructor for Strategy

Method Detail

defaultStrategy
```
public static Strategy defaultStrategy(String algo)
```
Construct a default set of parameters for DecisionTree

Parameters:
algo - "Classification" or "Regression"

algo
```
public scala.Enumeration.Value algo()
```

impurity
```
public Impurity impurity()
```

setImpurity

public void setImpurity(Impurity x$1)

maxDepth
```
public int maxDepth()
```

setMaxDepth
```
public void setMaxDepth(int x$1)
```

numClasses
```
public int numClasses()
```

setNumClasses
```
public void setNumClasses(int x$1)
```

maxBins
```
public int maxBins()
```

setMaxBins
```
public void setMaxBins(int x$1)
```

quantileCalculationStrategy

public scala.Enumeration.Value quantileCalculationStrategy()

setQuantileCalculationStrategy

public void setQuantileCalculationStrategy(scala.Enumeration.Value x$1)

categoricalFeaturesInfo

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

minInstancesPerNode
```
public int minInstancesPerNode()
```

setMinInstancesPerNode

public void setMinInstancesPerNode(int x$1)

minInfoGain
```
public double minInfoGain()
```

setMinInfoGain

public void setMinInfoGain(double x$1)

maxMemoryInMB
```
public int maxMemoryInMB()
```

setMaxMemoryInMB

public void setMaxMemoryInMB(int x$1)

subsamplingRate
```
public double subsamplingRate()
```

setSubsamplingRate

public void setSubsamplingRate(double x$1)

useNodeIdCache
```
public boolean useNodeIdCache()
```

setUseNodeIdCache

public void setUseNodeIdCache(boolean x$1)

checkpointDir

public scala.Option<String> checkpointDir()

setCheckpointDir

public void setCheckpointDir(scala.Option<String> x$1)

checkpointInterval
```
public int checkpointInterval()
```

setCheckpointInterval

public void setCheckpointInterval(int x$1)

isMulticlassClassification

public boolean isMulticlassClassification()

isMulticlassWithCategoricalFeatures

public boolean isMulticlassWithCategoricalFeatures()

setAlgo
```
public void setAlgo(String algo)
```
Sets Algorithm using a String.

setCategoricalFeaturesInfo

public void setCategoricalFeaturesInfo(java.util.Map<Integer,Integer> categoricalFeaturesInfo)

Sets categoricalFeaturesInfo using a Java Map.

assertValid
```
public void assertValid()
```
Check validity of parameters. Throws exception if invalid.

copy
```
public Strategy copy()
```
Returns a shallow copy of this instance.

getAlgo

public scala.Enumeration.Value getAlgo()

getImpurity
```
public Impurity getImpurity()
```

getMaxDepth
```
public int getMaxDepth()
```

getNumClasses
```
public int getNumClasses()
```

getMaxBins
```
public int getMaxBins()
```

getQuantileCalculationStrategy

public scala.Enumeration.Value getQuantileCalculationStrategy()

getCategoricalFeaturesInfo

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

getMinInstancesPerNode
```
public int getMinInstancesPerNode()
```

getMinInfoGain
```
public double getMinInfoGain()
```

getMaxMemoryInMB
```
public int getMaxMemoryInMB()
```

getSubsamplingRate
```
public double getSubsamplingRate()
```

getUseNodeIdCache
```
public boolean getUseNodeIdCache()
```

getCheckpointDir

public scala.Option<String> getCheckpointDir()

getCheckpointInterval
```
public int getCheckpointInterval()
```

Class Strategy

Constructor Summary

Method Summary

Methods inherited from class Object

Constructor Detail

Strategy

Strategy

Method Detail

defaultStrategy

algo

impurity

setImpurity

maxDepth

setMaxDepth

numClasses

setNumClasses

maxBins

setMaxBins

quantileCalculationStrategy

setQuantileCalculationStrategy

categoricalFeaturesInfo

minInstancesPerNode

setMinInstancesPerNode

minInfoGain

setMinInfoGain

maxMemoryInMB

setMaxMemoryInMB

subsamplingRate

setSubsamplingRate

useNodeIdCache

setUseNodeIdCache

checkpointDir

setCheckpointDir

checkpointInterval

setCheckpointInterval

isMulticlassClassification

isMulticlassWithCategoricalFeatures

setAlgo

setCategoricalFeaturesInfo

assertValid

copy

getAlgo

getImpurity

getMaxDepth

getNumClasses

getMaxBins

getQuantileCalculationStrategy

getCategoricalFeaturesInfo

getMinInstancesPerNode

getMinInfoGain

getMaxMemoryInMB

getSubsamplingRate

getUseNodeIdCache

getCheckpointDir

getCheckpointInterval