Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-10668

Use WeightedLeastSquares in LinearRegression with L2 regularization if the number of features is small

    XMLWordPrintableJSON

Details

    • New Feature
    • Status: Resolved
    • Critical
    • Resolution: Fixed
    • None
    • 1.6.0
    • ML
    • None

    Description

      If the number of features is small (<=4096) and the regularization is L2, we should use WeightedLeastSquares to solve the problem rather than L-BFGS. The former requires only one pass to the data.

      Attachments

        Issue Links

          Activity

            People

              lewuathe Kai
              mengxr Xiangrui Meng
              DB Tsai DB Tsai
              Votes:
              0 Vote for this issue
              Watchers:
              6 Start watching this issue

              Dates

                Created:
                Updated:
                Resolved: