ai.djl.mxnet.zoo.nlp.qa

Class MxBertVocabulary

java.lang.Object

ai.djl.modality.nlp.bert.BertVocabulary

ai.djl.mxnet.zoo.nlp.qa.MxBertVocabulary

All Implemented Interfaces:

ai.djl.modality.nlp.Vocabulary

public class MxBertVocabulary
extends ai.djl.modality.nlp.bert.BertVocabulary

A MXNet implementaion of BertVocabulary.

Constructor Summary

Constructors

Constructor and Description
MxBertVocabulary()

Method Summary

Modifier and Type	Method and Description
long	getIndex(java.lang.String token)
java.lang.String	getToken(long index)
static MxBertVocabulary	parse(java.io.InputStream is) Parses the vocabulary file and create MxBertVocabulary.

Methods inherited from class java.lang.Object

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

Constructor Detail

MxBertVocabulary

public MxBertVocabulary()

Method Detail

parse

public static MxBertVocabulary parse(java.io.InputStream is)

Parses the vocabulary file and create MxBertVocabulary.

Parameters:

is - the input InputStream of the vocabulary file

Returns:

an instance of MxBertVocabulary

getIndex

public long getIndex(java.lang.String token)

getToken

public java.lang.String getToken(long index)