Package ai.djl.fasttext.zoo.nlp.textclassification

Class TextClassificationModelLoader

java.lang.Object
ai.djl.repository.zoo.BaseModelLoader
ai.djl.fasttext.zoo.nlp.textclassification.TextClassificationModelLoader
All Implemented Interfaces:
ai.djl.repository.zoo.ModelLoader
public class TextClassificationModelLoader extends ai.djl.repository.zoo.BaseModelLoader
Model loader for fastText cooking stackexchange models.

  • Field Summary

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

    defaultFactory, mrl

  • Constructor Summary

    Constructors
    Constructor
    Description
    TextClassificationModelLoader(ai.djl.repository.Repository repository)
    Creates the Model loader from the given repository.

  • Method Summary

    Modifier and Type
    Method
    Description
    <I, O> ai.djl.repository.zoo.ZooModel<I,O>
    loadModel(ai.djl.repository.zoo.Criteria<I,O> criteria)

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

    createModel, getApplication, getArtifactId, getTranslatorFactory, listModels, toString

    Methods inherited from class java.lang.Object

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

  • Constructor Details

    • TextClassificationModelLoader

      public TextClassificationModelLoader(ai.djl.repository.Repository repository)
      Creates the Model loader from the given repository.
      Parameters:
      repository - the repository to load the model from

  • Method Details

    • loadModel

      public <I, O> ai.djl.repository.zoo.ZooModel<I,O> loadModel(ai.djl.repository.zoo.Criteria<I,O> criteria) throws ai.djl.repository.zoo.ModelNotFoundException, IOException, ai.djl.MalformedModelException
      Specified by:
      loadModel in interface ai.djl.repository.zoo.ModelLoader
      Overrides:
      loadModel in class ai.djl.repository.zoo.BaseModelLoader
      Throws:
      ai.djl.repository.zoo.ModelNotFoundException
      IOException
      ai.djl.MalformedModelException