Package 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	Description
BERTFeatureExtractor<?>	bert	BERTFeatureExtractor instance
Path	inputFile	Input file to read, one doc per line
Path	outputFile	Output json file.

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

header

Constructor Summary

Constructors

Constructor	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

getOptionsDescription

Field Details

bert

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

BERTFeatureExtractor instance

inputFile

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

Input file to read, one doc per line

outputFile

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

Output json file.

Constructor Details

BERTFeatureExtractorOptions

public BERTFeatureExtractorOptions()