diff --git a/src/Plugins/BotSharp.Plugin.RoutingSpeeder/Providers/IntentClassifier.cs b/src/Plugins/BotSharp.Plugin.RoutingSpeeder/Providers/IntentClassifier.cs index af6a892f..63d827b1 100644 --- a/src/Plugins/BotSharp.Plugin.RoutingSpeeder/Providers/IntentClassifier.cs +++ b/src/Plugins/BotSharp.Plugin.RoutingSpeeder/Providers/IntentClassifier.cs @@ -196,7 +196,10 @@ public class IntentClassifier public string[] GetLabels() { - return GetFiles().Select(x => Path.GetFileNameWithoutExtension(x)).ToArray(); + var agentService = _services.CreateScope().ServiceProvider.GetRequiredService(); + string rootDirectory = Path.Combine(agentService.GetDataDir(), _settings.RAW_DATA_DIR, _settings.LABEL_FILE_NAME); + var labelText = File.ReadAllLines(rootDirectory); + return labelText.OrderBy(x => x).ToArray(); } public string TextClean(string text) diff --git a/src/Plugins/BotSharp.Plugin.RoutingSpeeder/Settings/classifierSetting.cs b/src/Plugins/BotSharp.Plugin.RoutingSpeeder/Settings/classifierSetting.cs index d051f2e7..09bda6e8 100644 --- a/src/Plugins/BotSharp.Plugin.RoutingSpeeder/Settings/classifierSetting.cs +++ b/src/Plugins/BotSharp.Plugin.RoutingSpeeder/Settings/classifierSetting.cs @@ -15,4 +15,5 @@ public class ClassifierSetting public string RAW_DATA_DIR { get; set; } = "raw_data"; public string MODEL_DIR { get; set; } = "models"; + public string LABEL_FILE_NAME { get; set; } = "label.txt"; } diff --git a/src/WebStarter/data/models/intent-classifier.h5 b/src/WebStarter/data/models/intent-classifier.h5 new file mode 100644 index 00000000..e205dd6c Binary files /dev/null and b/src/WebStarter/data/models/intent-classifier.h5 differ diff --git a/src/WebStarter/data/raw_data/label.txt b/src/WebStarter/data/raw_data/label.txt new file mode 100644 index 00000000..9128ef85 --- /dev/null +++ b/src/WebStarter/data/raw_data/label.txt @@ -0,0 +1,3 @@ +goodbye +greeting +other \ No newline at end of file