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

DecisionTreeRunner functionality improvement

    XMLWordPrintableJSON

    Details

    • Type: Improvement
    • Status: Resolved
    • Priority: Minor
    • Resolution: Fixed
    • Affects Version/s: None
    • Fix Version/s: 1.2.0
    • Component/s: MLlib
    • Labels:
      None

      Description

      DecisionTreeRunner functionality additions:

      • Allow user to pass in a test dataset
      • Do not print full model if the model is too large.

      As part of this, modify DecisionTreeModel and RandomForestModel to allow printing less info. Proposed updates:

      • toString: prints model summary
      • toDebugString: prints full model (named after RDD.toDebugString)

      Similar update to Python API:

      • _repr_() now prints a model summary
      • toDebugString() now prints the full model

        Attachments

          Activity

            People

            • Assignee:
              josephkb Joseph K. Bradley
              Reporter:
              josephkb Joseph K. Bradley
            • Votes:
              0 Vote for this issue
              Watchers:
              3 Start watching this issue

              Dates

              • Created:
                Updated:
                Resolved: