[SONA] optimize pool capacity in lbfgs and owlqn

wodo2008 · Nov 23, 2017 · ea8c2fa · ea8c2fa
1 parent 771c507
commit ea8c2fa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/...el/mllib/src/main/scala/com/tencent/angel/spark/ml/classification/SparseLRWithLBFGS.scala b/...el/mllib/src/main/scala/com/tencent/angel/spark/ml/classification/SparseLRWithLBFGS.scala
@@ -94,7 +94,7 @@ object SparseLRWithLBFGS {
   }
 
   def runLBFGSAngel(trainData: RDD[(OneHotVector, Double)], dim: Int, m: Int, maxIter: Int): Unit = {
-    val initWeightPS = PSVector.dense(dim, 5 * m).toBreeze
+    val initWeightPS = PSVector.dense(dim, 3 * m).toBreeze
     val tol = 1e-6
     val lbfgs = new LBFGS[BreezePSVector](maxIter, m, tol)
     val states = lbfgs.iterations(SparseLogistic.PSCost(trainData), initWeightPS)

diff --git a/...el/mllib/src/main/scala/com/tencent/angel/spark/ml/classification/SparseLRWithOWLQN.scala b/...el/mllib/src/main/scala/com/tencent/angel/spark/ml/classification/SparseLRWithOWLQN.scala
@@ -96,12 +96,10 @@ object SparseLRWithOWLQN {
 
   def runPSOWLQN(trainData: RDD[(OneHotVector, Double)], dim: Int, m: Int, maxIter: Int): Unit = {
 
-    val initWeightPS = PSVector.dense(dim, 10 * m).toBreeze
-
+    val initWeightPS = PSVector.dense(dim, 3 * m).toBreeze
     val regPS = PSVector.duplicate(initWeightPS.component).toBreeze
 
     val tol = 1e-6
-
     val owlqn = new OWLQN(maxIter, m, regPS, tol)
     val states = owlqn.iterations(SparseLogistic.PSCost(trainData), initWeightPS)