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

Modify the functions

Browse source
This commit is contained in:
Wenbo Cao 2023-09-12 12:42:31 -05:00
parent eaab096ca0
commit 043980dee2
2 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/Plugins/BotSharp.Plugin.RoutingSpeeder/Controllers/RoutingSpeederController.cs
View file

@ -25,13 +25,12 @@ public class RoutingSpeederController : ControllerBase
public IActionResult TrainIntentClassifier(TrainingParams trainingParams)
{
var intentClassifier = _service.GetRequiredService<IntentClassifier>();
// intentClassifier.InitClassifer(trainingParams.Inference);
intentClassifier.Train(trainingParams);
return Ok(intentClassifier.Labels);
}
[HttpPost("/routing-speeder/classifier/inference")]
public IActionResult TrainIntentClassifier([FromBody] DialoguePredictionModel message)
public IActionResult InferenceIntentClassifier([FromBody] DialoguePredictionModel message)
{
var intentClassifier = _service.GetRequiredService<IntentClassifier>();
var vector = intentClassifier.GetTextEmbedding(message.Text);

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

@ -15,12 +15,14 @@ using Microsoft.Extensions.DependencyInjection;
using System.Linq;
using Tensorflow.Keras;
using BotSharp.Abstraction.Agents;
using Microsoft.Extensions.Logging;
namespace BotSharp.Plugin.RoutingSpeeder.Providers;
public class IntentClassifier
{
private readonly IServiceProvider _services;
private readonly ILogger _logger;
private KnowledgeBaseSettings _knowledgeBaseSettings;
Model _model;
public Model model => _model;
@ -31,11 +33,15 @@ public class IntentClassifier
private string[] _labels;
public string[] Labels => _labels == null ? GetLabels() : _labels;
public IntentClassifier(IServiceProvider services, ClassifierSetting settings, KnowledgeBaseSettings knowledgeBaseSettings)
public IntentClassifier(IServiceProvider services,
ClassifierSetting settings,
KnowledgeBaseSettings knowledgeBaseSettings,
ILogger logger)
{
_services = services;
_settings = settings;
_knowledgeBaseSettings = knowledgeBaseSettings;
_logger = logger;
}
private void Reset()
@ -121,7 +127,7 @@ public class IntentClassifier
{
var logInfo = _inferenceMode ? "No available weights." : "Will implement model training process and write trained weights into local";
_isModelReady = false;
Console.WriteLine(logInfo);
_logger.LogInformation(logInfo);
}
return weightsFile;