trianta/markov-generator
1
0
Fork 0
generated from trianta/cpp-unity-template
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed generation cutoff

Browse Source
This commit is contained in:
Trianta 2023-12-08 23:29:25 -06:00
parent 2eef3d36f4
commit cf56912436
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/generator.cpp
View File

@ -116,14 +116,7 @@ void Generator::Train(void) {
std::cout << "[Setup - Info] Begin training" << std::endl;
for (const auto& word : words) {
if (word.size() < setup.prefixLength) {
trie.Insert(word);
continue;
} else {
for (int i = 0; i < word.size() - setup.prefixLength; ++i) {
trie.Insert(word.substr(i, setup.prefixLength));
}
}
trie.Insert(word);
}
std::cout << "[Setup - Info] Finished training" << std::endl;
}