You are viewing a plain text version of this content. The canonical link for it is here.
Posted to issues@spark.apache.org by "Gang Bai (JIRA)" <ji...@apache.org> on 2014/06/17 11:39:06 UTC
[jira] [Created] (SPARK-2163) Change ``setConvergenceTol'' with a
parameter of type Double instead of Int
Gang Bai created SPARK-2163:
-------------------------------
Summary: Change ``setConvergenceTol'' with a parameter of type Double instead of Int
Key: SPARK-2163
URL: https://issues.apache.org/jira/browse/SPARK-2163
Project: Spark
Issue Type: Improvement
Components: MLlib
Affects Versions: 1.0.0
Reporter: Gang Bai
The class LBFGS in mllib.optimization currently provides a {{setConvergenceTol(tolerance: Int)}} method for setting the convergence tolerance. The tolerance parameter is of type {{Int}}. The specified tolerance is then used as parameter in calling {{LBFGS.runLBFGS}}, where the parameter {{convergenceTol}} is of type {{Double}}.
The Int parameter may cause problem when one creates an optimizer and sets a Double-valued tolerance. e.g:
{code:borderStyle=solid}
override val optimizer = new LBFGS(gradient, updater)
.setNumCorrections(9)
.setConvergenceTol(1e-4) // *type mismatch here*
.setMaxNumIterations(100)
.setRegParam(1.0)
{code}
IMHO there is no need to make the tolerance of type Int. Let's change it into a Double parameter and eliminate the type mismatch problem.
--
This message was sent by Atlassian JIRA
(v6.2#6252)