smile.classification

Class GradientTreeBoost.Trainer

java.lang.Object

smile.classification.ClassifierTrainer<double[]>

smile.classification.GradientTreeBoost.Trainer

Enclosing class:

GradientTreeBoost

public static class GradientTreeBoost.Trainer
extends ClassifierTrainer<double[]>

Trainer for GradientTreeBoost classifiers.

Constructor Summary

Constructors

Constructor and Description
Trainer() Default constructor of 500 trees.
Trainer(Attribute[] attributes, int T) Constructor.
Trainer(int T) Constructor.

Method Summary

Modifier and Type	Method and Description
GradientTreeBoost.Trainer	setMaximumLeafNodes(int J) Sets the maximum number of leaf nodes in the tree.
GradientTreeBoost.Trainer	setNumTrees(int T) Sets the number of trees in the random forest.
GradientTreeBoost.Trainer	setSamplingRates(double f) Sets the sampling rate for stochastic tree boosting.
GradientTreeBoost.Trainer	setShrinkage(double shrinkage) Sets the shrinkage parameter in (0, 1] controls the learning rate of procedure.
GradientTreeBoost	train(double[][] x, int[] y) Learns a classifier with given training data.

Methods inherited from class smile.classification.ClassifierTrainer

setAttributes

Methods inherited from class java.lang.Object

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

Constructor Detail

Trainer

public Trainer()

Default constructor of 500 trees.

Trainer

public Trainer(int T)

Constructor.

Parameters:

T - the number of trees.

Trainer

public Trainer(Attribute[] attributes,
               int T)

Constructor.

Parameters:

attributes - the attributes of independent variable.

T - the number of trees.

Method Detail

setNumTrees

public GradientTreeBoost.Trainer setNumTrees(int T)

Sets the number of trees in the random forest.

Parameters:

T - the number of trees.

setMaximumLeafNodes

public GradientTreeBoost.Trainer setMaximumLeafNodes(int J)

Sets the maximum number of leaf nodes in the tree.

Parameters:

J - the maximum number of leaf nodes in the tree.

setShrinkage

public GradientTreeBoost.Trainer setShrinkage(double shrinkage)

Sets the shrinkage parameter in (0, 1] controls the learning rate of procedure.

Parameters:

shrinkage - the learning rate.

setSamplingRates

public GradientTreeBoost.Trainer setSamplingRates(double f)

Sets the sampling rate for stochastic tree boosting.

Parameters:

f - the sampling rate for stochastic tree boosting.

train

public GradientTreeBoost train(double[][] x,
                               int[] y)

Description copied from class: ClassifierTrainer

Learns a classifier with given training data.

Specified by:

train in class ClassifierTrainer<double[]>

Parameters:

x - the training instances.

y - the training labels.

Returns:

a trained classifier.