Class: Strategy

eclairjs/mllib/tree/configuration.Strategy

new Strategy(algo, impurity, maxDepth, numClasses, maxBins, quantileCalculationStrategy, categoricalFeaturesInfo, minInstancesPerNode, minInfoGain, maxMemoryInMB, subsamplingRate, useNodeIdCache, checkpointInterval)

Parameters:
Name Type Description
algo Algo
impurity Impurity
maxDepth number
numClasses number
maxBins number
quantileCalculationStrategy QuantileStrategy
categoricalFeaturesInfo Map
minInstancesPerNode number
minInfoGain number
maxMemoryInMB number
subsamplingRate number
useNodeIdCache boolean
checkpointInterval number
Source:

Methods

setCategoricalFeaturesInfo(categoricalFeaturesInfo) → {Promise.<Void>}

Sets categoricalFeaturesInfo using a Java Map.
Parameters:
Name Type Description
categoricalFeaturesInfo Map
Source:
Returns:
A Promise that resolves to nothing.
Type
Promise.<Void>

setMaxDepth(num) → {Promise.<void>}

Parameters:
Name Type Description
num integer
Source:
Returns:
Type
Promise.<void>

setNumClasses(num) → {Promise.<void>}

Parameters:
Name Type Description
num integer
Source:
Returns:
Type
Promise.<void>