Package org.gridgain.ml.model

Enum Class ModelType

java.lang.Object

java.lang.Enum<ModelType>

org.gridgain.ml.model.ModelType

All Implemented Interfaces:

Serializable, Comparable<ModelType>, Constable

public enum ModelType
extends Enum<ModelType>

Enum defining the types of machine learning models supported by the framework.

Nested Class Summary

Nested classes/interfaces inherited from class java.lang.Enum

Enum.EnumDesc<E extends Enum<E>>

Enum Constant Summary

Enum Constants

Enum Constant	Description
ONNX	ONNX models.
PYTORCH	PyTorch models (TorchScript format).
TENSORFLOW	TensorFlow models (SavedModel format).

Method Summary

Modifier and Type	Method	Description
static ModelType	valueOf(String name)	Returns the enum constant of this class with the specified name.
static ModelType[]	values()	Returns an array containing the constants of this enum class, in the order they are declared.

Methods inherited from class java.lang.Enum

clone, compareTo, describeConstable, equals, finalize, getDeclaringClass, hashCode, name, ordinal, toString, valueOf

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Enum Constant Details

PYTORCH

public static final ModelType PYTORCH

PyTorch models (TorchScript format).

TENSORFLOW

public static final ModelType TENSORFLOW

TensorFlow models (SavedModel format).

ONNX

public static final ModelType ONNX

ONNX models.

Method Details

values

public static ModelType[] values()

Returns an array containing the constants of this enum class, in the order they are declared.

Returns:

an array containing the constants of this enum class, in the order they are declared

valueOf

public static ModelType valueOf(String name)

Returns the enum constant of this class with the specified name. The string must match exactly an identifier used to declare an enum constant in this class. (Extraneous whitespace characters are not permitted.)

Parameters:

name - the name of the enum constant to be returned.

Returns:

the enum constant with the specified name

Throws:

IllegalArgumentException - if this enum class has no constant with the specified name

NullPointerException - if the argument is null