org.tribuo.interop.onnx.extractors

Class BERTFeatureExtractor.BERTFeatureExtractorOptions

java.lang.Object

org.tribuo.interop.onnx.extractors.BERTFeatureExtractor.BERTFeatureExtractorOptions

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Options

Enclosing class:

BERTFeatureExtractor<T extends Output<T>>

public static class BERTFeatureExtractor.BERTFeatureExtractorOptions
extends Object
implements com.oracle.labs.mlrg.olcut.config.Options

CLI options for running BERT.

Field Summary

Fields

Modifier and Type	Field and Description
BERTFeatureExtractor<?>	bert
Path	inputFile
Path	outputFile

Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options

header

Constructor Summary

Constructors

Constructor and Description
BERTFeatureExtractorOptions()

Method Summary

Methods inherited from class java.lang.Object

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

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

formatUsage, generateTypeDescription, getAllOptions, getEnumConstantString, getOptionFields, getOptions, getOptionsDescription, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

bert

@Option(charName=98,
        longName="bert",
        usage="BERTFeatureExtractor instance")
public BERTFeatureExtractor<?> bert

inputFile

@Option(charName=105,
        longName="input-file",
        usage="Input file to read, one doc per line")
public Path inputFile

outputFile

@Option(charName=111,
        longName="output-file",
        usage="Output json file.")
public Path outputFile

Constructor Detail

BERTFeatureExtractorOptions

public BERTFeatureExtractorOptions()