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
Path	inputFile
Path	outputFile

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

inputFile

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

outputFile

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

Constructor Details

BERTFeatureExtractorOptions

public BERTFeatureExtractorOptions()