NLClassifier

Classifier API para tarefas de classificação de linguagem natural, categoriza string em diferentes classes.

A API espera um modelo TFLite com o seguinte tensor de entrada / saída:

• Tensor de entrada (kTfLiteString)
  • entrada do modelo, aceita uma string.
• Tensor de pontuação de saída (kTfLiteUInt8 / kTfLiteInt8 / kTfLiteInt16 / kTfLiteFloat32 / kTfLiteFloat64 / kTfLiteBool)
  • resultados de saída para cada classe, se tipo for um dos tipos Int, desquantize-o; se for do tipo Bool, converta os valores para 0,0 e 1,0 respectivamente.
  • pode ter um arquivo opcional associado em metadados para rótulos, o arquivo deve ser um arquivo de texto simples com um rótulo por linha, o número de rótulos deve corresponder ao número de categorias que o modelo produz. Tensor do rótulo de saída: opcional (kTfLiteString) - nome da classe de saída para cada classe, deve ter o mesmo comprimento com pontuações. Se esse tensor não estiver presente, a API usa índices de pontuação como nomes de classe. - será ignorado se o tensor de pontuação de saída já tiver um arquivo de etiqueta associado.
• Tensor de rótulo de saída opcional (kTfLiteString / kTfLiteInt32)
  • output classname para cada classe, deve ter o mesmo comprimento com pontuações. Se esse tensor não estiver presente, a API usa índices de pontuação como nomes de classe.
  • será ignorado se o tensor de pontuação de saída já tiver um arquivo labe associado.

Por padrão as tentativas de API para encontrar os tensores de entrada / saída com configurações padrão no NLClassifier.NLClassifierOptions , com tensor nome prioridade sobre tensor índice. A opção é configurável para diferentes modelos TFLite.