mirror of
https://github.com/Findus23/se-simulator.git
synced 2024-09-19 15:53:45 +02:00
21 lines
527 B
Python
21 lines
527 B
Python
import markovify
|
|
from nltk.tokenize.moses import MosesDetokenizer, MosesTokenizer
|
|
|
|
tokenizer = MosesTokenizer()
|
|
detokenizer = MosesDetokenizer()
|
|
|
|
|
|
class MarkovText(markovify.Text):
|
|
def word_split(self, sentence):
|
|
return tokenizer.tokenize(sentence)
|
|
|
|
def word_join(self, words):
|
|
return detokenizer.detokenize(words, return_str=True)
|
|
|
|
|
|
class MarkovUserName(markovify.Text):
|
|
def word_split(self, word):
|
|
return list(word)
|
|
|
|
def word_join(self, characters):
|
|
return "".join(characters)
|