change param

ynqa · Dec 8, 2020 · d0b729d · d0b729d
1 parent f6c558a
commit d0b729d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/e2e.go b/test/e2e.go
@@ -105,9 +105,9 @@ func main() {
 				glove.BatchSize(10000),
 				glove.Dim(50),
 				glove.Goroutines(20),
-				glove.Initlr(0.01),
+				glove.Initlr(0.03),
 				glove.Iter(3),
-				glove.MinCount(10),
+				glove.MinCount(20),
 				glove.Solver(glove.Stochastic),
 				glove.Verbose(),
 				glove.Window(10),
@@ -119,9 +119,9 @@ func main() {
 				glove.BatchSize(10000),
 				glove.Dim(50),
 				glove.Goroutines(20),
-				glove.Initlr(0.01),
+				glove.Initlr(0.03),
 				glove.Iter(3),
-				glove.MinCount(10),
+				glove.MinCount(20),
 				glove.Solver(glove.AdaGrad),
 				glove.Verbose(),
 				glove.Window(10),
@@ -133,7 +133,7 @@ func main() {
 				lexvec.BatchSize(10000),
 				lexvec.Dim(50),
 				lexvec.Goroutines(20),
-				lexvec.Iter(1),
+				lexvec.Iter(3),
 				lexvec.MinCount(10),
 				lexvec.Relation(lexvec.PPMI),
 				lexvec.Verbose(),