SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Merge pull request #51 from fabioparra/master

Browse source 
BotSharpNER.predict should loop CRFSegOut array
This commit is contained in:
Haiping Chen 2019-11-27 15:52:23 -06:00 committed by GitHub
parent ba18c77bc1 722e6482fe
commit b6604324f8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
BotSharp.Core/Engines/BotSharp/BotSharpNER.cs
View file

@ -189,15 +189,24 @@ namespace BotSharp.Core.Engines.BotSharp
for (int i = 0; i < sent.Tokens.Count; i++)
{
var entity = crf_out[0].result_;
entities.Add(new NlpEntity
for (int crf_index = 0; crf_index < crf_out.Length; crf_index++)
{
Entity = entity[i],
Start = doc.Sentences[0].Tokens[i].Start,
Value = doc.Sentences[0].Tokens[i].Text,
Confidence = 0,
Extrator = "BotSharpNER"
});
//entity can be extract in more than 1 run/index
//var entity = crf_out[0].result_;
var entity = crf_out[crf_index].result_;
var contains = entities.Exists(x => x.Entity == entity[i]);
if (!contains)
{
entities.Add(new NlpEntity
{
Entity = entity[i],
Start = doc.Sentences[0].Tokens[i].Start,
Value = doc.Sentences[0].Tokens[i].Text,
Confidence = 0,
Extrator = "BotSharpNER"
});
}
}
}
sent.Entities = MergeEntity(doc.Sentences[0].Text, entities);