Package opennlp.dl.doccat

# Class DocumentCategorizerDL

java.lang.Object
opennlp.dl.AbstractDL
opennlp.dl.doccat.DocumentCategorizerDL

All Implemented Interfaces:
  `AutoCloseable`, `opennlp.tools.doccat.DocumentCategorizer`

public class DocumentCategorizerDL extends AbstractDL implements opennlp.tools.doccat.DocumentCategorizer
An implementation of `DocumentCategorizer` that performs document classification using ONNX models.

See Also:
  - `DocumentCategorizer`
  - `InferenceOptions`
  - `ClassificationScoringStrategy`

- ## Field Summary

  ### Fields inherited from class opennlp.dl.AbstractDL

  `ATTENTION_MASK, INPUT_IDS, TOKEN_TYPE_IDS`

- ## Constructor Summary

  Constructors
  Constructor
  Description
  `DocumentCategorizerDL(File model, File vocabulary, File config, ClassificationScoringStrategy classificationScoringStrategy, InferenceOptions inferenceOptions)`
  Instantiates a `document categorizer` using ONNX models.
  `DocumentCategorizerDL(File model, File vocabulary, Map<Integer,String> categories, ClassificationScoringStrategy classificationScoringStrategy, InferenceOptions inferenceOptions)`
  Instantiates a `document categorizer` using ONNX models.

- ## Method Summary

  Modifier and Type
  Method
  Description
  `double[]`
  `categorize(String[] strings)`
  `double[]`
  `categorize(String[] strings, Map<String,Object> map)`
  `String`
  `getAllResults(double[] doubles)`
  `String`
  `getBestCategory(double[] doubles)`
  `String`
  `getCategory(int i)`
  `int`
  `getIndex(String s)`
  `int`
  `getNumberOfCategories()`
  `Map<String,Double>`
  `scoreMap(String[] strings)`
  `SortedMap<Double,Set<String>>`
  `sortedScoreMap(String[] strings)`

  ### Methods inherited from class opennlp.dl.AbstractDL

  `close, loadVocab`

  ### Methods inherited from class java.lang.Object

  `equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait`

- ## Constructor Details

  - ### DocumentCategorizerDL

    public DocumentCategorizerDL(File model, File vocabulary, Map<Integer,String> categories, ClassificationScoringStrategy classificationScoringStrategy, InferenceOptions inferenceOptions) throws IOException, ai.onnxruntime.OrtException
    Instantiates a `document categorizer` using ONNX models.
    Parameters:
      `model` - The ONNX model file.
      `vocabulary` - The model file's vocabulary file.
      `categories` - The categories.
      `classificationScoringStrategy` - Implementation of `ClassificationScoringStrategy` used to calculate the classification scores given the score of each individual document part.
      `inferenceOptions` - `InferenceOptions` to control the inference.
    Throws:
      `ai.onnxruntime.OrtException` - Thrown if the `model` cannot be loaded.
      `IOException` - Thrown if errors occurred loading the `model` or `vocabulary`.

  - ### DocumentCategorizerDL

    public DocumentCategorizerDL(File model, File vocabulary, File config, ClassificationScoringStrategy classificationScoringStrategy, InferenceOptions inferenceOptions) throws IOException, ai.onnxruntime.OrtException
    Instantiates a `document categorizer` using ONNX models.
    Parameters:
      `model` - The ONNX model file.
      `vocabulary` - The model file's vocabulary file.
      `config` - The model's config file. The file will be used to determine the classification categories.
      `classificationScoringStrategy` - Implementation of `ClassificationScoringStrategy` used to calculate the classification scores given the score of each individual document part.
      `inferenceOptions` - `InferenceOptions` to control the inference.
    Throws:
      `ai.onnxruntime.OrtException` - Thrown if the `model` cannot be loaded.
      `IOException` - Thrown if errors occurred loading the `model` or `vocabulary`.

- ## Method Details

  - ### categorize

    public double[] categorize(String[] strings)
    Specified by:
      `categorize` in interface `opennlp.tools.doccat.DocumentCategorizer`

  - ### categorize

    public double[] categorize(String[] strings, Map<String,Object> map)
    Specified by:
      `categorize` in interface `opennlp.tools.doccat.DocumentCategorizer`

  - ### getBestCategory

    public String getBestCategory(double[] doubles)
    Specified by:
      `getBestCategory` in interface `opennlp.tools.doccat.DocumentCategorizer`

  - ### getIndex

    public int getIndex(String s)
    Specified by:
      `getIndex` in interface `opennlp.tools.doccat.DocumentCategorizer`

  - ### getCategory

    public String getCategory(int i)
    Specified by:
      `getCategory` in interface `opennlp.tools.doccat.DocumentCategorizer`

  - ### getNumberOfCategories

    public int getNumberOfCategories()
    Specified by:
      `getNumberOfCategories` in interface `opennlp.tools.doccat.DocumentCategorizer`

  - ### getAllResults

    public String getAllResults(double[] doubles)
    Specified by:
      `getAllResults` in interface `opennlp.tools.doccat.DocumentCategorizer`

  - ### scoreMap

    public Map<String,Double> scoreMap(String[] strings)
    Specified by:
      `scoreMap` in interface `opennlp.tools.doccat.DocumentCategorizer`

  - ### sortedScoreMap

    public SortedMap<Double,Set<String>> sortedScoreMap(String[] strings)
    Specified by:
      `sortedScoreMap` in interface `opennlp.tools.doccat.DocumentCategorizer`