org.apache.ignite.ml.selection.paramgrid

Class ParamGrid

java.lang.Object

org.apache.ignite.ml.selection.paramgrid.ParamGrid

public class ParamGrid
extends Object

Keeps the grid of parameters.

Constructor Summary

Constructors

Constructor and Description
ParamGrid()

Method Summary

Modifier and Type	Method and Description
ParamGrid	addHyperParam(String paramName, DoubleConsumer setter, Double[] params) Adds a grid for the specific hyper parameter.
HyperParameterTuningStrategy	getHyperParameterTuningStrategy() Returns the Hyper-parameter tuning strategy.
String	getParamNameByIndex(int idx) Returns the name of hyper-parameter by the given index.
List<Double[]>	getParamRawData() Prepare data for hyper-parameter tuning.
Map<Integer,Double[]>	getParamValuesByParamIdx()
DoubleConsumer	getSetterByIndex(int idx) Returns setter for parameter with the given index.
ParamGrid	withParameterSearchStrategy(HyperParameterTuningStrategy paramSearchStgy) Set up the hyperparameter searching strategy.

Methods inherited from class java.lang.Object

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

Constructor Detail

ParamGrid

public ParamGrid()

Method Detail

getParamValuesByParamIdx

public Map<Integer,Double[]> getParamValuesByParamIdx()

addHyperParam

public ParamGrid addHyperParam(String paramName,
                               DoubleConsumer setter,
                               Double[] params)

Adds a grid for the specific hyper parameter.

Parameters:

paramName - The parameter name.

setter - The method reference to trainer or preprocessor trainer setter.

params - The array of the given hyper parameter values.

Returns:

The updated ParamGrid.

withParameterSearchStrategy

public ParamGrid withParameterSearchStrategy(HyperParameterTuningStrategy paramSearchStgy)

Set up the hyperparameter searching strategy.

Parameters:

paramSearchStgy - Parameter search strategy.

getHyperParameterTuningStrategy

public HyperParameterTuningStrategy getHyperParameterTuningStrategy()

Returns the Hyper-parameter tuning strategy.

getSetterByIndex

public DoubleConsumer getSetterByIndex(int idx)

Returns setter for parameter with the given index.

getParamNameByIndex

public String getParamNameByIndex(int idx)

Returns the name of hyper-parameter by the given index.

getParamRawData

public List<Double[]> getParamRawData()

Prepare data for hyper-parameter tuning.