Uses of Class
org.tribuo.util.onnx.ONNXNode
Package
Description
Provides the core interfaces and classes for using Tribuo.
Provides majority vote ensemble combiners for classification
along with an implementation of multiclass Adaboost.
Provides an interface to LibLinear-java for classification problems.
Provides an interface to LibSVM for classification problems.
Provides an implementation of a classification factorization machine using Stochastic Gradient Descent.
Provides an implementation of a classification linear model using Stochastic Gradient Descent.
Provides the base classes for models trained with stochastic gradient descent.
Provides an interface for model prediction combinations,
two base classes for ensemble models, a base class for
ensemble excuses, and a Bagging implementation.
Provides math related util classes.
Provides a multi-label ensemble combiner that performs a (possibly
weighted) majority vote among each label independently, along with an
implementation of classifier chain ensembles.
Provides an implementation of a multi-label classification factorization machine model using Stochastic Gradient Descent.
Provides an implementation of a multi-label classification linear model using Stochastic Gradient Descent.
Provides
EnsembleCombiner
implementations
for working with multi-output regression problems.Provides an interface to liblinear for regression problems.
Provides an interface to LibSVM for regression problems.
Provides an implementation of factorization machines for regression using Stochastic Gradient Descent.
Provides an implementation of linear regression using Stochastic Gradient Descent.
Provides implementations of sparse linear regression using various forms of regularisation penalty.
Interfaces and utilities for writing ONNX models from Java.
-
Uses of ONNXNode in org.tribuo
-
Uses of ONNXNode in org.tribuo.classification.ensemble
Modifier and TypeMethodDescriptionFullyWeightedVotingCombiner.exportCombiner
(ONNXNode input) Exports this voting combiner to ONNX.FullyWeightedVotingCombiner.exportCombiner
(ONNXNode input, T weight) Exports this voting combiner to ONNX.VotingCombiner.exportCombiner
(ONNXNode input) Exports this voting combiner to ONNX.VotingCombiner.exportCombiner
(ONNXNode input, T weight) Exports this voting combiner to ONNXModifier and TypeMethodDescriptionFullyWeightedVotingCombiner.exportCombiner
(ONNXNode input) Exports this voting combiner to ONNX.FullyWeightedVotingCombiner.exportCombiner
(ONNXNode input, T weight) Exports this voting combiner to ONNX.VotingCombiner.exportCombiner
(ONNXNode input) Exports this voting combiner to ONNX.VotingCombiner.exportCombiner
(ONNXNode input, T weight) Exports this voting combiner to ONNX -
Uses of ONNXNode in org.tribuo.classification.liblinear
-
Uses of ONNXNode in org.tribuo.classification.libsvm
-
Uses of ONNXNode in org.tribuo.classification.sgd.fm
Modifier and TypeMethodDescriptionprotected ONNXNode
FMClassificationModel.onnxOutput
(ONNXNode input) Modifier and TypeMethodDescriptionprotected ONNXNode
FMClassificationModel.onnxOutput
(ONNXNode input) -
Uses of ONNXNode in org.tribuo.classification.sgd.linear
-
Uses of ONNXNode in org.tribuo.common.sgd
Modifier and TypeMethodDescriptionprotected abstract ONNXNode
AbstractFMModel.onnxOutput
(ONNXNode input) Takes the unnormalized ONNX output of this model and applies an appropriate normalizer from the concrete class.protected abstract ONNXNode
AbstractLinearSGDModel.onnxOutput
(ONNXNode input) Takes the unnormalized ONNX output of this model and applies an appropriate normalizer from the concrete class.AbstractFMModel.writeONNXGraph
(ONNXRef<?> input) AbstractLinearSGDModel.writeONNXGraph
(ONNXRef<?> input) Modifier and TypeMethodDescriptionprotected abstract ONNXNode
AbstractFMModel.onnxOutput
(ONNXNode input) Takes the unnormalized ONNX output of this model and applies an appropriate normalizer from the concrete class.protected abstract ONNXNode
AbstractLinearSGDModel.onnxOutput
(ONNXNode input) Takes the unnormalized ONNX output of this model and applies an appropriate normalizer from the concrete class. -
Uses of ONNXNode in org.tribuo.ensemble
Modifier and TypeMethodDescriptiondefault ONNXNode
EnsembleCombiner.exportCombiner
(ONNXNode input) Exports this ensemble combiner into the ONNX context of its input.EnsembleCombiner.exportCombiner
(ONNXNode input, U weight) Exports this ensemble combiner into the ONNX context of its input.WeightedEnsembleModel.writeONNXGraph
(ONNXRef<?> input) Modifier and TypeMethodDescriptiondefault ONNXNode
EnsembleCombiner.exportCombiner
(ONNXNode input) Exports this ensemble combiner into the ONNX context of its input.EnsembleCombiner.exportCombiner
(ONNXNode input, U weight) Exports this ensemble combiner into the ONNX context of its input. -
Uses of ONNXNode in org.tribuo.math.util
Modifier and TypeMethodDescriptionExpNormalizer.exportNormalizer
(ONNXNode input) Returns the ONNX softmax node over the 2nd dimension.NoopNormalizer.exportNormalizer
(ONNXNode input) Returns its input.Normalizer.exportNormalizer
(ONNXNode input) Applies ONNX ReduceMin, Sub, ReduceSum, and Div operations to input.SigmoidNormalizer.exportNormalizer
(ONNXNode input) Returns the ONNX sigmoid node, operating independently over each element.default ONNXNode
VectorNormalizer.exportNormalizer
(ONNXNode input) Exports this normalizer to ONNX, returning the leaf of the appended graph and writing the nodes needed for normalization into theONNXContext
thatinput
belongs to.Modifier and TypeMethodDescriptionExpNormalizer.exportNormalizer
(ONNXNode input) Returns the ONNX softmax node over the 2nd dimension.NoopNormalizer.exportNormalizer
(ONNXNode input) Returns its input.Normalizer.exportNormalizer
(ONNXNode input) Applies ONNX ReduceMin, Sub, ReduceSum, and Div operations to input.SigmoidNormalizer.exportNormalizer
(ONNXNode input) Returns the ONNX sigmoid node, operating independently over each element.default ONNXNode
VectorNormalizer.exportNormalizer
(ONNXNode input) Exports this normalizer to ONNX, returning the leaf of the appended graph and writing the nodes needed for normalization into theONNXContext
thatinput
belongs to. -
Uses of ONNXNode in org.tribuo.multilabel.ensemble
Modifier and TypeMethodDescriptionMultiLabelVotingCombiner.exportCombiner
(ONNXNode input) Exports this voting combiner to ONNX.MultiLabelVotingCombiner.exportCombiner
(ONNXNode input, T weight) Exports this voting combiner to ONNXModifier and TypeMethodDescriptionMultiLabelVotingCombiner.exportCombiner
(ONNXNode input) Exports this voting combiner to ONNX.MultiLabelVotingCombiner.exportCombiner
(ONNXNode input, T weight) Exports this voting combiner to ONNX -
Uses of ONNXNode in org.tribuo.multilabel.sgd.fm
Modifier and TypeMethodDescriptionprotected ONNXNode
FMMultiLabelModel.onnxOutput
(ONNXNode input) Modifier and TypeMethodDescriptionprotected ONNXNode
FMMultiLabelModel.onnxOutput
(ONNXNode input) -
Uses of ONNXNode in org.tribuo.multilabel.sgd.linear
-
Uses of ONNXNode in org.tribuo.regression.ensemble
Modifier and TypeMethodDescriptionAveragingCombiner.exportCombiner
(ONNXNode input) Exports this averaging combiner, writing constructed nodes into theONNXContext
governinginput
and returning the leaf node of the combiner.AveragingCombiner.exportCombiner
(ONNXNode input, T weight) Exports this averaging combiner, writing constructed nodes into theONNXContext
governinginput
and returning the leaf node of the combiner.Modifier and TypeMethodDescriptionAveragingCombiner.exportCombiner
(ONNXNode input) Exports this averaging combiner, writing constructed nodes into theONNXContext
governinginput
and returning the leaf node of the combiner.AveragingCombiner.exportCombiner
(ONNXNode input, T weight) Exports this averaging combiner, writing constructed nodes into theONNXContext
governinginput
and returning the leaf node of the combiner. -
Uses of ONNXNode in org.tribuo.regression.liblinear
-
Uses of ONNXNode in org.tribuo.regression.libsvm
-
Uses of ONNXNode in org.tribuo.regression.sgd.fm
Modifier and TypeMethodDescriptionprotected ONNXNode
FMRegressionModel.onnxOutput
(ONNXNode fmOutput) Modifier and TypeMethodDescriptionprotected ONNXNode
FMRegressionModel.onnxOutput
(ONNXNode fmOutput) -
Uses of ONNXNode in org.tribuo.regression.sgd.linear
-
Uses of ONNXNode in org.tribuo.regression.slm
-
Uses of ONNXNode in org.tribuo.util.onnx
Modifier and TypeMethodDescriptionONNXRef.apply
(ONNXOperator op) Convenience method that callsONNXContext.operation(ONNXOperator, List, String)
, using this ONNXRef as the argument toinputs
.ONNXRef.apply
(ONNXOperator op, String outputName) Convenience method that callsONNXContext.operation(ONNXOperator, List, String)
, using this ONNXRef as the argument toinputs
.ONNXRef.apply
(ONNXOperator op, List<ONNXRef<?>> others) Convenience method that callsONNXContext.operation(ONNXOperator, List, String, Map)
, using this ONNXRef as the first argument toinputs
, withotherInputs
append as subsequent arguments.ONNXRef.apply
(ONNXOperator op, List<ONNXRef<?>> others, String outputName) Convenience method that callsONNXContext.operation(ONNXOperator, List, String, Map)
, using this ONNXRef as the argument toinputs
, withotherInputs
append as subsequent arguments.ONNXRef.apply
(ONNXOperator op, Map<String, Object> attributes) Convenience method that callsONNXContext.operation(ONNXOperator, List, String, Map)
, using this ONNXRef as the argument toinputs
.ONNXRef.apply
(ONNXOperator op, ONNXRef<?> other) Convenience method that callsONNXContext.operation(ONNXOperator, List, String, Map)
, passing this ONNXRef andother
as a length 2 list toinputs
.ONNXRef.apply
(ONNXOperator op, ONNXRef<?> other, String outputName) Convenience method that callsONNXContext.operation(ONNXOperator, List, String)
, passing this ONNXRef andother
as a length 2 list toinputs
.Convenience method that callsONNXContext.operation(ONNXOperator, List, String, Map)
, passing this ONNXRef andother
as a length 2 list toinputs
.Casts this ONNXRef to a different type using theONNXOperators.CAST
operation, and returning the output node of that op.ONNXContext.operation
(ONNXOperator op, List<T> inputs, String outputName) Method for creatingONNXNode
s fromONNXOperator
instances and inputs.ONNXContext.operation
(ONNXOperator op, List<T> inputs, String outputName, Map<String, Object> attributes) Method for creatingONNXNode
s fromONNXOperator
and inputs.Modifier and TypeMethodDescriptionConvenience method that callsONNXContext.operation(ONNXOperator, List, List, Map)
, using this ONNXRef as the argument toinputs
.ONNXRef.apply
(ONNXOperator op, List<ONNXRef<?>> otherInputs, List<String> outputs, Map<String, Object> attributes) Convenience method that callsONNXContext.operation(ONNXOperator, List, List, Map)
, using this ONNXRef as the first argument toinputs
, withotherInputs
append as subsequent arguments.ONNXContext.operation
(ONNXOperator op, List<T> inputs, List<String> outputs, Map<String, Object> attributes) Base method for creatingONNXNode
s fromONNXOperator
and inputs.