ai.djl.mxnet.zoo.nlp.embedding

Class GloveWordEmbeddingModelLoader

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>

ai.djl.mxnet.zoo.nlp.embedding.GloveWordEmbeddingModelLoader

All Implemented Interfaces:

ai.djl.repository.zoo.ModelLoader<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>

public class GloveWordEmbeddingModelLoader
extends ai.djl.repository.zoo.BaseModelLoader<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>

A ModelLoader for a WordEmbedding based on GloVe.

Field Summary

Fields inherited from class ai.djl.repository.zoo.BaseModelLoader

factories, modelZoo, mrl, repository, version

Constructor Summary

Constructors

Constructor and Description
GloveWordEmbeddingModelLoader(ai.djl.repository.Repository repository) Constructs a GloveWordEmbeddingModelLoader given the repository.

Method Summary

Modifier and Type	Method and Description
protected ai.djl.Model	createModel(java.lang.String name, ai.djl.Device device, ai.djl.repository.Artifact artifact, java.util.Map<java.lang.String,java.lang.Object> arguments, java.lang.String engine)
ai.djl.Application	getApplication()
ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>	loadModel(java.util.Map<java.lang.String,java.lang.String> filters, ai.djl.Device device, ai.djl.util.Progress progress)

Methods inherited from class ai.djl.repository.zoo.BaseModelLoader

getArtifactId, listModels, loadModel, match, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.repository.zoo.ModelLoader

loadModel, loadModel, loadModel, loadModel

Constructor Detail

GloveWordEmbeddingModelLoader

public GloveWordEmbeddingModelLoader(ai.djl.repository.Repository repository)

Constructs a GloveWordEmbeddingModelLoader given the repository.

Parameters:

repository - the repository to load the model from

Method Detail

getApplication

public ai.djl.Application getApplication()

createModel

protected ai.djl.Model createModel(java.lang.String name,
                                   ai.djl.Device device,
                                   ai.djl.repository.Artifact artifact,
                                   java.util.Map<java.lang.String,java.lang.Object> arguments,
                                   java.lang.String engine)
                            throws java.io.IOException

Overrides:

createModel in class ai.djl.repository.zoo.BaseModelLoader<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>

Throws:

java.io.IOException

loadModel

public ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> loadModel(java.util.Map<java.lang.String,java.lang.String> filters,
                                                                                             ai.djl.Device device,
                                                                                             ai.djl.util.Progress progress)
                                                                                      throws java.io.IOException,
                                                                                             ai.djl.repository.zoo.ModelNotFoundException,
                                                                                             ai.djl.MalformedModelException

Throws:

java.io.IOException

ai.djl.repository.zoo.ModelNotFoundException

ai.djl.MalformedModelException