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

Add confidence threshold

Browse source
This commit is contained in:
Wenbo Cao 2023-08-31 20:42:04 -05:00
parent 1ba2e2674e
commit fc96c36dc0
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/Plugins/BotSharp.Plugin.RoutingSpeeder/Providers/IntentClassifier.cs
View file

@ -210,19 +210,23 @@ public class IntentClassifier
return processedText;
}
public string Predict(NDArray vector)
public string Predict(NDArray vector, float confidenceScore = 0.9f)
{
if (!_isModelReady)
{
InitClassifer();
}
var prob = _model.predict(vector);
var prob = _model.predict(vector).numpy();
if (prob[0] < confidenceScore)
{
return string.Empty;
}
var probLabel = tf.arg_max(prob, -1).numpy();
// var prediction = _settings.LabelMappingDict.First(x => x.Value == probLabel[0]).Key;
var prediction = GetLabels()[probLabel[0]];
// var prediction = GetLabels().Where((x, i) => i == probLabel[0]).First();
return prediction;
}