diff options
| -rw-r--r-- | ebooks.cpp | 8 | ||||
| -rw-r--r-- | gen.cpp | 8 |
2 files changed, 8 insertions, 8 deletions
| diff --git a/ebooks.cpp b/ebooks.cpp index 8b567ed..52247da 100644 --- a/ebooks.cpp +++ b/ebooks.cpp | |||
| @@ -64,14 +64,14 @@ int main(int argc, char** args) | |||
| 64 | rawr kgramstats; | 64 | rawr kgramstats; |
| 65 | kgramstats.addCorpus(corpus); | 65 | kgramstats.addCorpus(corpus); |
| 66 | kgramstats.compile(4); | 66 | kgramstats.compile(4); |
| 67 | kgramstats.setTransformCallback([&] (std::string canonical, std::string) { | 67 | kgramstats.setTransformCallback([&] (std::string, std::string form) { |
| 68 | size_t pos = canonical.find("$name$"); | 68 | size_t pos = form.find("$name$"); |
| 69 | if (pos != std::string::npos) | 69 | if (pos != std::string::npos) |
| 70 | { | 70 | { |
| 71 | canonical.replace(pos, 6, fv_names[rand() % fv_names.size()]); | 71 | form.replace(pos, 6, fv_names[rand() % fv_names.size()]); |
| 72 | } | 72 | } |
| 73 | 73 | ||
| 74 | return canonical; | 74 | return form; |
| 75 | }); | 75 | }); |
| 76 | 76 | ||
| 77 | std::mutex stats_mutex; | 77 | std::mutex stats_mutex; |
| diff --git a/gen.cpp b/gen.cpp index eba0277..5e2d9db 100644 --- a/gen.cpp +++ b/gen.cpp | |||
| @@ -69,14 +69,14 @@ int main(int argc, char** args) | |||
| 69 | rawr kgramstats; | 69 | rawr kgramstats; |
| 70 | kgramstats.addCorpus(corpus); | 70 | kgramstats.addCorpus(corpus); |
| 71 | kgramstats.compile(4); | 71 | kgramstats.compile(4); |
| 72 | kgramstats.setTransformCallback([&] (std::string canonical, std::string) { | 72 | kgramstats.setTransformCallback([&] (std::string, std::string form) { |
| 73 | size_t pos = canonical.find("$name$"); | 73 | size_t pos = form.find("$name$"); |
| 74 | if (pos != std::string::npos) | 74 | if (pos != std::string::npos) |
| 75 | { | 75 | { |
| 76 | canonical.replace(pos, 6, fv_names[rand() % fv_names.size()]); | 76 | form.replace(pos, 6, fv_names[rand() % fv_names.size()]); |
| 77 | } | 77 | } |
| 78 | 78 | ||
| 79 | return canonical; | 79 | return form; |
| 80 | }); | 80 | }); |
| 81 | 81 | ||
| 82 | std::cout << "Generating..." << std::endl; | 82 | std::cout << "Generating..." << std::endl; |
