public class BoostingStrategy
extends Object
implements scala.Serializable, scala.Product
GradientBoostedTrees
.
Constructor and Description |
---|
BoostingStrategy(Strategy treeStrategy,
Loss loss,
int numIterations,
double learningRate) |
Modifier and Type | Method and Description |
---|---|
void |
assertValid()
Check validity of parameters.
|
static BoostingStrategy |
defaultParams(scala.Enumeration.Value algo)
Returns default configuration for the boosting algorithm
|
static BoostingStrategy |
defaultParams(String algo)
Returns default configuration for the boosting algorithm
|
double |
getLearningRate() |
Loss |
getLoss() |
int |
getNumIterations() |
Strategy |
getTreeStrategy() |
double |
learningRate() |
Loss |
loss() |
int |
numIterations() |
void |
setLearningRate(double x$1) |
void |
setLoss(Loss x$1) |
void |
setNumIterations(int x$1) |
void |
setTreeStrategy(Strategy x$1) |
Strategy |
treeStrategy() |
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
public static BoostingStrategy defaultParams(String algo)
algo
- Learning goal. Supported: "Classification" or "Regression"public static BoostingStrategy defaultParams(scala.Enumeration.Value algo)
algo
- Learning goal. Supported:
org.apache.spark.mllib.tree.configuration.Algo.Classification
,
org.apache.spark.mllib.tree.configuration.Algo.Regression
public Strategy treeStrategy()
public void setTreeStrategy(Strategy x$1)
public Loss loss()
public void setLoss(Loss x$1)
public int numIterations()
public void setNumIterations(int x$1)
public double learningRate()
public void setLearningRate(double x$1)
public void assertValid()
public Strategy getTreeStrategy()
public Loss getLoss()
public int getNumIterations()
public double getLearningRate()