org.tribuo.interop.onnx

Class DenseTransformer

java.lang.Object

org.tribuo.interop.onnx.DenseTransformer

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Configurable, com.oracle.labs.mlrg.olcut.provenance.Provenancable<com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance>, Serializable, ExampleTransformer

public class DenseTransformer
extends Object
implements ExampleTransformer

Converts a sparse Tribuo example into a dense float vector, then wraps it in an OnnxTensor.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
static int	THRESHOLD Feature size beyond which a warning is generated (as ONNX requires dense features and large feature spaces are memory hungry).
static int	WARNING_THRESHOLD Number of times the feature size warning should be printed.

Constructor Summary

Constructors

Constructor and Description
DenseTransformer()

Method Summary

Modifier and Type	Method and Description
com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance	getProvenance()
String	toString()
ai.onnxruntime.OnnxTensor	transform(ai.onnxruntime.OrtEnvironment env, List<SparseVector> vectors) Converts a list of SparseVectors representing a batch of features into a OnnxTensor.
ai.onnxruntime.OnnxTensor	transform(ai.onnxruntime.OrtEnvironment env, SparseVector vector) Converts a SparseVector representing the features into a OnnxTensor.

Methods inherited from class java.lang.Object

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

Methods inherited from interface com.oracle.labs.mlrg.olcut.config.Configurable

postConfig

Field Detail

THRESHOLD

public static final int THRESHOLD

Feature size beyond which a warning is generated (as ONNX requires dense features and large feature spaces are memory hungry).

See Also:

Constant Field Values

WARNING_THRESHOLD

public static final int WARNING_THRESHOLD

Number of times the feature size warning should be printed.

See Also:

Constant Field Values

Constructor Detail

DenseTransformer

public DenseTransformer()

Method Detail

transform

public ai.onnxruntime.OnnxTensor transform(ai.onnxruntime.OrtEnvironment env,
                                           SparseVector vector)
                                    throws ai.onnxruntime.OrtException

Description copied from interface: ExampleTransformer

Converts a SparseVector representing the features into a OnnxTensor.

It generates it as a single example minibatch.

Specified by:

transform in interface ExampleTransformer

Parameters:

env - The OrtEnvironment to create the tensor in.

vector - The features to convert.

Returns:

A dense OnnxTensor representing this vector.

Throws:

ai.onnxruntime.OrtException - if the transformation failed.

transform

public ai.onnxruntime.OnnxTensor transform(ai.onnxruntime.OrtEnvironment env,
                                           List<SparseVector> vectors)
                                    throws ai.onnxruntime.OrtException

Description copied from interface: ExampleTransformer

Converts a list of SparseVectors representing a batch of features into a OnnxTensor.

Specified by:

transform in interface ExampleTransformer

Parameters:

env - The OrtEnvironment to create the tensor in.

vectors - The batch of features to convert.

Returns:

A dense OnnxTensor representing this minibatch.

Throws:

ai.onnxruntime.OrtException - if the transformation failed.

toString

public String toString()

Overrides:

toString in class Object

getProvenance

public com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance getProvenance()

Specified by:

getProvenance in interface com.oracle.labs.mlrg.olcut.provenance.Provenancable<com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance>