Update test_cpp.cpp

bab2min · Dec 17, 2024 · 63efcad · 63efcad
1 parent 2389067
commit 63efcad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_cpp.cpp b/test/test_cpp.cpp
@@ -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)
 	{
 		{