From 3fc8ab98f1212daa0bc20fe92ac7cab0c3bccd94 Mon Sep 17 00:00:00 2001 From: Nikhil Kansal Date: Thu, 4 Jan 2018 00:07:01 -0800 Subject: [PATCH] add method to parse with probabilities --- lib/utterance_parser/parser.rb | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/lib/utterance_parser/parser.rb b/lib/utterance_parser/parser.rb index 8705e76..7de3230 100644 --- a/lib/utterance_parser/parser.rb +++ b/lib/utterance_parser/parser.rb @@ -47,6 +47,13 @@ def parse(text) [intent, extract_entities(labeled)] end + def parse_with_probabilities(text) + utterance = Utterance.new(text) + intents = @classifier.classify(utterance.pos_tokens) + labeled = @labeller.label([ utterance.pos_tokens.map { |t| t.join(" ") } ]).first + [intents, extract_entities(labeled)] + end + def save(path=nil) build_save_paths path if path