From 6e7f060c7d0cd07259ce0ab6e28d31e890d0cc61 Mon Sep 17 00:00:00 2001
From: Bolo <36342729+PppBr@users.noreply.github.com>
Date: Fri, 10 Aug 2018 10:42:00 -0500
Subject: [PATCH] finish the Predict API by putting the predict result in
 pipeline

---
 .../Engines/CRFsuite/CRFsuiteEntityRecognizer.cs    | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/BotSharp.Core/Engines/CRFsuite/CRFsuiteEntityRecognizer.cs b/BotSharp.Core/Engines/CRFsuite/CRFsuiteEntityRecognizer.cs
index fb06cd778..399e95f6a 100644
--- a/BotSharp.Core/Engines/CRFsuite/CRFsuiteEntityRecognizer.cs
+++ b/BotSharp.Core/Engines/CRFsuite/CRFsuiteEntityRecognizer.cs
@@ -187,6 +187,19 @@ public async Task<bool> Predict(Agent agent, JObject data, PipeModel meta)
 
             var entities = new List<NlpEntity>();
             //
+            string[] entityProbabilityPairs = output.Split("\r");
+            for (int i = 0 ; i < entityProbabilityPairs.Length ; i++)
+            {
+                string entityProbabilityPair = entityProbabilityPairs[i];
+                string entity = entityProbabilityPair.Split(":")[0];
+                decimal probability = decimal.Parse(entityProbabilityPair.Split(":")[1]);
+                NlpEntity nlpentity = new NlpEntity();
+                nlpentity.Entity = entity;
+                nlpentity.Value = tokens[0][i].Text;
+                nlpentity.Confidence = probability;
+                entities.Add(nlpentity);
+            }
+            
 
 
             data["entities"] = JObject.FromObject(entities);