Gavin老师Transformer直播课感悟 - 基于Transformer的Rasa Internals解密之框架DIETClassifier及TED(十一)

        本文继续围绕工业级业务对话平台和框架Rasa解析DAG图中两个核心组件DIETClassifier和TEDPolicy的内部工作机制。

一、DIETClassifier

1. 源码介绍

        DIETClassifier作为DAG图中的核心组件，实现了GraphComponent接口，DIET作为一个多任务的模型，能够同时完成意图分类和实体提取这两大任务，DIETClassifier继承了IntentClassifier和EntityExtractorMixin。

IntentClassifier目前没有具体实现，这是为了将来可以进行扩展：

EntityExtractorMixin提供了实体提取需要使用的一些公共的方法：