Package com.yahoo.searchdefinition.processing

Class OnnxModelConfigGenerator

java.lang.Object

com.yahoo.searchdefinition.processing.Processor

com.yahoo.searchdefinition.processing.OnnxModelConfigGenerator

public class OnnxModelConfigGenerator
extends Processor

Processes ONNX ranking features of the form: onnx("files/model.onnx", "path/to/output:1") And generates an "onnx-model" configuration as if it was defined in the schema: onnx-model files_model_onnx { file: "files/model.onnx" } Inputs and outputs are resolved in OnnxModelTypeResolver, which must be processed after this.

Author:

lesters

Field Summary

Fields inherited from class com.yahoo.searchdefinition.processing.Processor

deployLogger, queryProfiles, rankProfileRegistry, schema

Constructor Summary

Constructors

Constructor	Description
OnnxModelConfigGenerator(Schema schema, com.yahoo.config.application.api.DeployLogger deployLogger, RankProfileRegistry rankProfileRegistry, QueryProfiles queryProfiles)

Method Summary

Modifier and Type	Method	Description
void	process(boolean validate, boolean documentsOnly)	Processes the input search definition by modifying the input search and its documents, and returns the input search definition.

Methods inherited from class com.yahoo.searchdefinition.processing.Processor

addField, fail, formatError, info, info, matchingRankSettingsIterator, newProcessException, newProcessException, process, warn, warn

Methods inherited from class java.lang.Object

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

Constructor Detail

OnnxModelConfigGenerator

public OnnxModelConfigGenerator(Schema schema,
                                com.yahoo.config.application.api.DeployLogger deployLogger,
                                RankProfileRegistry rankProfileRegistry,
                                QueryProfiles queryProfiles)

Method Detail

process

public void process(boolean validate,
                    boolean documentsOnly)

Description copied from class: Processor

Processes the input search definition by modifying the input search and its documents, and returns the input search definition.

Specified by:

process in class Processor

Parameters:

validate - true to throw exceptions on validation errors, false to make the best possible effort at completing processing without throwing an exception. If we are not validating, emitting warnings have no effect and can (but must not) be skipped.

documentsOnly - true to skip processing (including validation, regardless of the validate setting) of aspects not relating to document definitions (e.g rank profiles)