Package org.tribuo.util.onnx

package org.tribuo.util.onnx

Interfaces and utilities for writing ONNX models from Java.

Developed to support Tribuo, but can be used to export other machine learning models from JVM languages.

Related Packages

Package	Description
org.tribuo.util	Provides utilities which don't have other Tribuo dependencies.
org.tribuo.util.infotheory	This package provides static classes of information theoretic functions.
org.tribuo.util.tokens	Core definitions for tokenization.

Class	Description
ONNXAttribute	The spec for an attribute, used to produce the attribute proto at construction time.
ONNXContext	Context object used to scope and manage the creation of ONNX OnnxMl.GraphProto and OnnxMl.ModelProto instances.
ONNXInitializer	A subclass of ONNXRef specialized for OnnxMl.TensorProto.
ONNXNode	A subclass of ONNXRef specialized for OnnxMl.NodeProto.
ONNXOperators	The supported ONNX operators.
ONNXPlaceholder	A subclass of ONNXRef specialized for OnnxMl.ValueInfoProto.
ONNXRef<T extends com.google.protobuf.GeneratedMessageV3>	An abstract reference that represents both a node in an ONNX computation graph and a container for a specific ONNX proto object that denotes that node.
ONNXUtils	Helper functions for building ONNX protos.