Package ai.djl.mxnet.zoo.nlp.embedding.utils

Class BuildModelZooWordEmbedding

java.lang.Object

ai.djl.mxnet.zoo.nlp.embedding.utils.BuildModelZooWordEmbedding

public final class BuildModelZooWordEmbedding
extends Object

A utility to build embeddings into DJL format from Gluon-NLP format.

This utility should be called after running convertEmbeddings.py

Method Summary

Modifier and Type	Method	Description
static void	main(String[] args)	Runs main build embeddings after editing.

Methods inherited from class java.lang.Object

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

Method Details

main

public static void main(String[] args)
                 throws IOException

Runs main build embeddings after editing.

Parameters:

args - the arguments

Throws:

IOException - thrown if unable to read files in directory