コード例 #1
0
from kochat.proc import DistanceClassifier, GensimEmbedder, EntityRecognizer, SoftmaxClassifier

from konlpy.tag import Okt as Mecab

from sprinkle.scenarios.scenarios import call, schedule

dataset = Dataset(ood=True)
emb = GensimEmbedder(model=embed.FastText())

clf = DistanceClassifier(
    model=intent.CNN(dataset.intent_dict),
    loss=CenterLoss(dataset.intent_dict),
)

rcn = EntityRecognizer(
    model=entity.LSTM(dataset.entity_dict),
    loss=CRFLoss(dataset.entity_dict)
)

kochat = KochatApi(
    dataset=dataset,
    embed_processor=(emb, False),
    intent_classifier=(clf, False),
    entity_recognizer=(rcn, False),
    scenarios=[
        call, schedule
    ]
)

class VoiceCommandController():
    def __init__(self):
コード例 #2
0
from sprinkle.scenarios.scenarios import call, schedule
from sprinkle.controllers.UserController import UserController
from sprinkle.controllers.VoiceCommandController import VoiceCommandController

userController = UserController()
voiceCommandControler = VoiceCommandController()

dataset = Dataset(ood=True)
emb = GensimEmbedder(model=embed.FastText())

clf = DistanceClassifier(
    model=intent.CNN(dataset.intent_dict),
    loss=CenterLoss(dataset.intent_dict),
)

rcn = EntityRecognizer(model=entity.LSTM(dataset.entity_dict),
                       loss=CRFLoss(dataset.entity_dict))

kochat = KochatApi(dataset=dataset,
                   embed_processor=(emb, False),
                   intent_classifier=(clf, False),
                   entity_recognizer=(rcn, False),
                   scenarios=[call, schedule])

# @kochat.app.route('/')
# def index():
#     # userController.searchUserById("1234567890")
#     return "adsf"


@kochat.app.route('/users/init', methods=['POST'])