Details

    • Type: Sub-task
    • Status: Open
    • Priority: Minor
    • Resolution: Unresolved
    • Affects Version/s: 3.1.0
    • Fix Version/s: None
    • Component/s: ML
    • Labels:
      None

      Description

      performace test in https://issues.apache.org/jira/browse/SPARK-31783,
      Huber loss seems start to diverge since 70 iters.

       

       for (size <- Seq(1, 4, 16, 64); iter <- Seq(10, 50, 100)) {
          Thread.sleep(10000)
          val hlir = new LinearRegression().setLoss("huber").setSolver("l-bfgs").setMaxIter(iter).setTol(0)
          val start = System.currentTimeMillis
          val model = hlir.setBlockSize(size).fit(df)
          val end = System.currentTimeMillis
          println((model.uid, size, iter, end - start, model.summary.objectiveHistory.last, model.summary.totalIterations, model.coefficients.toString.take(100)))
      }

        Attachments

        1. huber.xlsx
          65 kB
          zhengruifeng

          Issue Links

            Activity

              People

              • Assignee:
                Unassigned
                Reporter:
                podongfeng zhengruifeng
              • Votes:
                0 Vote for this issue
                Watchers:
                1 Start watching this issue

                Dates

                • Created:
                  Updated: