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) public IActionResult TrainIntentClassifier(TrainingParams trainingParams)
{ {
var intentClassifier = _service.GetRequiredService<IntentClassifier>(); var intentClassifier = _service.GetRequiredService<IntentClassifier>();
// intentClassifier.InitClassifer(trainingParams.Inference);
intentClassifier.Train(trainingParams); intentClassifier.Train(trainingParams);
return Ok(intentClassifier.Labels); return Ok(intentClassifier.Labels);
} }
[HttpPost("/routing-speeder/classifier/inference")] [HttpPost("/routing-speeder/classifier/inference")]
public IActionResult TrainIntentClassifier([FromBody] DialoguePredictionModel message) public IActionResult InferenceIntentClassifier([FromBody] DialoguePredictionModel message)
{ {
var intentClassifier = _service.GetRequiredService<IntentClassifier>(); var intentClassifier = _service.GetRequiredService<IntentClassifier>();
var vector = intentClassifier.GetTextEmbedding(message.Text); 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 System.Linq;
using Tensorflow.Keras; using Tensorflow.Keras;
using BotSharp.Abstraction.Agents; using BotSharp.Abstraction.Agents;
using Microsoft.Extensions.Logging;
namespace BotSharp.Plugin.RoutingSpeeder.Providers; namespace BotSharp.Plugin.RoutingSpeeder.Providers;
public class IntentClassifier public class IntentClassifier
{ {
private readonly IServiceProvider _services; private readonly IServiceProvider _services;
private readonly ILogger _logger;
private KnowledgeBaseSettings _knowledgeBaseSettings; private KnowledgeBaseSettings _knowledgeBaseSettings;
Model _model; Model _model;
public Model model => _model; public Model model => _model;
@ -31,11 +33,15 @@ public class IntentClassifier
private string[] _labels; private string[] _labels;
public string[] Labels => _labels == null ? GetLabels() : _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; _services = services;
_settings = settings; _settings = settings;
_knowledgeBaseSettings = knowledgeBaseSettings; _knowledgeBaseSettings = knowledgeBaseSettings;
_logger = logger;
} }
private void Reset() 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"; var logInfo = _inferenceMode ? "No available weights." : "Will implement model training process and write trained weights into local";
_isModelReady = false; _isModelReady = false;
Console.WriteLine(logInfo); _logger.LogInformation(logInfo);
} }
return weightsFile; return weightsFile;