Update test_cpp.cpp

This commit is contained in:
bab2min 2024-12-17 23:08:20 +09:00
commit 63efcade56

View file

@ -413,7 +413,7 @@ TEST(KiwiCpp, HSDataset)
for (size_t w : {0, 1, 2, 4})
{
//std::cout << w << std::endl;
auto dataset = kw.makeHSDataset(data, batchSize, 0, windowSize, w, 0.);
auto dataset = kw.makeHSDataset(data, batchSize, 0, windowSize, w, 0., 0.);
for (size_t i = 0; i < 2; ++i)
{
size_t totalBatchCnt = 0, totalTokenCnt = 0, s;
@ -436,7 +436,7 @@ TEST(KiwiCpp, HSDataset)
};
HSDataset trainset, devset;
trainset = kw.makeHSDataset(data, batchSize, 0, windowSize, 1, 0., tokenFilter, {}, 0.1, false, {}, 0, &devset);
trainset = kw.makeHSDataset(data, batchSize, 0, windowSize, 1, 0., 0., tokenFilter, {}, 0.1, false, {}, 0, &devset);
for (size_t i = 0; i < 2; ++i)
{
{