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 java.lang.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 and Description
static void	main(java.lang.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 Detail

main

public static void main(java.lang.String[] args)
                 throws java.io.IOException

Runs main build embeddings after editing.

Parameters:

args - the arguments

Throws:

java.io.IOException - thrown if unable to read files in directory