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 Predict(Agent agent, JObject data, PipeModel meta) var entities = new List(); // + 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);