aboutsummaryrefslogtreecommitdiff
path: root/mllib/src
diff options
context:
space:
mode:
authorlewuathe <lewuathe@me.com>2015-03-17 12:11:57 -0700
committerXiangrui Meng <meng@databricks.com>2015-03-17 12:11:57 -0700
commitd9f3e01688ad0a8d5fc2419a948a682ad7d957c9 (patch)
treef2f254a28aebcf6de03b2e88f1486804a7736531 /mllib/src
parent4cca3917dc30ee907e6cbd6a569b6ac58af963f7 (diff)
downloadspark-d9f3e01688ad0a8d5fc2419a948a682ad7d957c9.tar.gz
spark-d9f3e01688ad0a8d5fc2419a948a682ad7d957c9.tar.bz2
spark-d9f3e01688ad0a8d5fc2419a948a682ad7d957c9.zip
[SPARK-6336] LBFGS should document what convergenceTol means
LBFGS uses convergence tolerance. This value should be written in document as an argument. Author: lewuathe <lewuathe@me.com> Closes #5033 from Lewuathe/SPARK-6336 and squashes the following commits: e738b33 [lewuathe] Modify text to be more natural ac03c3a [lewuathe] Modify documentations 6ccb304 [lewuathe] [SPARK-6336] LBFGS should document what convergenceTol means
Diffstat (limited to 'mllib/src')
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala6
1 files changed, 5 insertions, 1 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala b/mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala
index d5e4f4ccbf..ef6eccd907 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala
@@ -60,6 +60,8 @@ class LBFGS(private var gradient: Gradient, private var updater: Updater)
/**
* Set the convergence tolerance of iterations for L-BFGS. Default 1E-4.
* Smaller value will lead to higher accuracy with the cost of more iterations.
+ * This value must be nonnegative. Lower convergence values are less tolerant
+ * and therefore generally cause more iterations to be run.
*/
def setConvergenceTol(tolerance: Double): this.type = {
this.convergenceTol = tolerance
@@ -142,7 +144,9 @@ object LBFGS extends Logging {
* one single data example)
* @param updater - Updater function to actually perform a gradient step in a given direction.
* @param numCorrections - The number of corrections used in the L-BFGS update.
- * @param convergenceTol - The convergence tolerance of iterations for L-BFGS
+ * @param convergenceTol - The convergence tolerance of iterations for L-BFGS which is must be
+ * nonnegative. Lower values are less tolerant and therefore generally
+ * cause more iterations to be run.
* @param maxNumIterations - Maximal number of iterations that L-BFGS can be run.
* @param regParam - Regularization parameter
*