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

Incorrect threshould length in 'setThresholds()' evoke Exception

Log workAgile BoardRank to TopRank to BottomAttach filesAttach ScreenshotBulk Copy AttachmentsBulk Move AttachmentsVotersWatch issueWatchersCreate sub-taskConvert to sub-taskMoveLinkCloneLabelsUpdate Comment AuthorReplace String in CommentUpdate Comment VisibilityDelete CommentsDelete
    XMLWordPrintableJSON

Details

    • Improvement
    • Status: Resolved
    • Trivial
    • Resolution: Fixed
    • None
    • 2.1.0
    • ML
    • None

    Description

      val path = "./spark-2.0.0-bin-hadoop2.7/data/mllib/sample_multiclass_classification_data.txt"
      val data = spark.read.format("libsvm").load(path)
      val rf = new RandomForestClassifier()
      val model = rf.fit(data)
      
      model.numClasses
      res48: Int = 3
      
      model.setThresholds(Array(0.5,0.1))
      res49: org.apache.spark.ml.classification.RandomForestClassificationModel = RandomForestClassificationModel (uid=rfc_b39da354ac8b) with 20 trees
      
      
      model.transform(data)
      java.lang.IllegalArgumentException: requirement failed: RandomForestClassificationModel.transform() called with non-matching numClasses and thresholds.length. numClasses=3, but thresholds has length 2
        at scala.Predef$.require(Predef.scala:224)
        at org.apache.spark.ml.classification.ProbabilisticClassificationModel.transform(ProbabilisticClassifier.scala:101)
        ... 58 elided
      

      Although model set with wrong threshoulds will fail in prediction, it maybe nice to evoke exception earlier in setThreshoulds

      Attachments

        Activity

          This comment will be Viewable by All Users Viewable by All Users
          Cancel

          People

            podongfeng Ruifeng Zheng Assign to me
            podongfeng Ruifeng Zheng
            Votes:
            0 Vote for this issue
            Watchers:
            2 Start watching this issue

            Dates

              Created:
              Updated:
              Resolved:

              Slack

                Issue deployment