Package org.gridgain.ml.model

Class ModelConfig

java.lang.Object

org.gridgain.ml.model.ModelConfig

All Implemented Interfaces:

Serializable

public class ModelConfig
extends Object
implements Serializable

Class representing configuration settings for a machine learning model.

See Also:

• Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	ModelConfig.Builder	Builder class for creating ModelConfig instances.

Method Summary

Modifier and Type	Method	Description
int	batchSize()	Gets the batch size.
static ModelConfig.Builder	builder()	Creates a new builder for ModelConfig.
Map<String,Object>	engineOptions()	Gets the engine options.

Methods inherited from class java.lang.Object

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

Method Details

batchSize

public int batchSize()

Gets the batch size.

Returns:

Batch size

engineOptions

public Map<String,Object> engineOptions()

Gets the engine options.

Returns:

Engine options

builder

public static ModelConfig.Builder builder()

Creates a new builder for ModelConfig.

Returns:

A new Builder instance