ai.djl.pytorch.zoo.nlp.sentimentanalysis

Class DistilBertSentimentAnalysisModelLoader

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader

ai.djl.pytorch.zoo.nlp.sentimentanalysis.DistilBertSentimentAnalysisModelLoader

All Implemented Interfaces:

ai.djl.repository.zoo.ModelLoader

public class DistilBertSentimentAnalysisModelLoader
extends ai.djl.repository.zoo.BaseModelLoader

Model loader for PyTorch DistilBERT Sentiment Analysis models.

The model was trained with PyTorch from https://github.com/huggingface/transformers.

See the BERT paper

See Also:

PtSymbolBlock

Field Summary

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

factories, modelZoo, resource

Constructor Summary

Constructors

Constructor and Description
DistilBertSentimentAnalysisModelLoader(ai.djl.repository.Repository repository) Creates the Model loader from the given repository.

Method Summary

Modifier and Type	Method and Description
ai.djl.repository.zoo.ZooModel<java.lang.String,ai.djl.modality.Classifications>	loadModel(java.util.Map<java.lang.String,java.lang.String> filters, ai.djl.Device device, ai.djl.util.Progress progress) Loads the model with the given search filters.

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

createModel, getApplication, getArtifactId, listModels, loadModel, toString

Methods inherited from class java.lang.Object

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

Constructor Detail

DistilBertSentimentAnalysisModelLoader

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

Creates the Model loader from the given repository.

Parameters:

repository - the repository to load the model from

Method Detail

loadModel

public ai.djl.repository.zoo.ZooModel<java.lang.String,ai.djl.modality.Classifications> 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

Loads the model with the given search filters.

Parameters:

filters - the search filters to match against the loaded model

device - the device the loaded model should use

progress - the progress tracker to update while loading the model

Returns:

the loaded model

Throws:

java.io.IOException - for various exceptions loading data from the repository

ai.djl.repository.zoo.ModelNotFoundException - if no model with the specified criteria is found

ai.djl.MalformedModelException - if the model data is malformed