代码之家  ›  专栏  ›  技术社区  ›  KOB

是否可以加载经过训练的Rasa-NLU模型并运行推理以获得嵌入

  •  0
  • KOB  · 技术社区  · 5 年前

    我训练了一个 supervised_embeddings Rasa NLU模型使用 rasa train nlu 对于我的特定用例,我现在需要获取输入的用户消息的嵌入,并将其与NLU训练数据中的消息嵌入进行比较。

    是否可以使用Python API来加载经过训练的模型,并使用它来获得文本字符串的嵌入?

    0 回复  |  直到 5 年前
        1
  •  3
  •   Melinda    5 年前

    这是在 rasa forum 为便于参考,请通过 only_output_properties=False interpreter.parse 方法:

    您可以在python脚本中执行此操作:

    from rasa.nlu.model import Interpreter
    
    interpreter = Interpreter.load('models/nlu-xyz/nlu')) ## this should be an extracted model
    result = interpreter.parse('hello world',only_output_properties=False)
    embeds = result.get("text_sparse_features")
    
        2
  •  0
  •   dragster    5 年前

    @KOB很遗憾,我们没有API。您可以在 https://github.com/RasaHQ/rasa/blob/master/rasa/nlu/classifiers/embedding_intent_classifier.py#L572 和使用 message_embed 作为嵌入。 请随时提出任何进一步的问题。