BotSharp/BotSharp.NLP/Tokenize/TokenizerFactory.cs
2018-09-06 17:32:51 -05:00

58 lines
1.6 KiB
C#

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace BotSharp.NLP.Tokenize
{
/// <summary>
/// BotSharp Tokenizer Factory
/// Tokenizers divide strings into lists of substrings.
/// The particular tokenizer requires implement interface
/// models to be installed.BotSharp.NLP also provides a simpler, regular-expression based tokenizer, which splits text on whitespace and punctuation.
/// </summary>
public class TokenizerFactory<ITokenize> where ITokenize : ITokenizer, new()
{
private SupportedLanguage _lang;
private ITokenize _tokenizer;
private TokenizationOptions _options;
public TokenizerFactory(TokenizationOptions options, SupportedLanguage lang)
{
_lang = lang;
_options = options;
_tokenizer = new ITokenize();
}
public List<Token> Tokenize(string sentence)
{
return _tokenizer.Tokenize(sentence, _options);
}
public List<List<Token>> Tokenize(List<String> sentences)
{
var sents = sentences.Select(s => new ParallelToken { Text = s }).ToList();
Parallel.ForEach(sents, (sentence) =>
{
sentence.Tokens = Tokenize(sentence.Text);
});
List<List<Token>> result = new List<List<Token>>();
sents.ForEach(x => result.Add(x.Tokens));
return result;
}
private class ParallelToken
{
public String Text { get; set; }
public List<Token> Tokens { get; set; }
}
}
}