Package org.gridgain.ml.model

Class ModelConfig.Builder

java.lang.Object

org.gridgain.ml.model.ModelConfig.Builder

Enclosing class:

ModelConfig

public static class ModelConfig.Builder
extends Object

Builder class for creating ModelConfig instances.

Constructor Summary

Constructors

Constructor	Description
Builder()

Method Summary

Modifier and Type	Method	Description
ModelConfig.Builder	batchSize(int batchSize)	Sets the batch size.
ModelConfig	build()	Builds the ModelConfig instance.
ModelConfig.Builder	from(ModelConfig config)	Creates a builder based on an existing ModelConfig.

Methods inherited from class java.lang.Object

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

Constructor Details

Builder

public Builder()

Method Details

batchSize

public ModelConfig.Builder batchSize(int batchSize)

Sets the batch size.

Parameters:

batchSize - Batch size

Returns:

Builder instance

from

public ModelConfig.Builder from(ModelConfig config)

Creates a builder based on an existing ModelConfig.

Parameters:

config - The base ModelConfig

Returns:

A builder initialized with values from the ModelConfig

build

public ModelConfig build()

Builds the ModelConfig instance.

Returns:

ModelConfig instance