Scala DecisionTreeModel类代码示例

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Scala›Scala教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

本文整理汇总了Scala中org.apache.spark.mllib.tree.model.DecisionTreeModel类的典型用法代码示例。如果您正苦于以下问题：Scala DecisionTreeModel类的具体用法？Scala DecisionTreeModel怎么用？Scala DecisionTreeModel使用的例子？那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。

在下文中一共展示了DecisionTreeModel类的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Scala代码示例。

示例1:

//设置package包名称以及导入依赖的类
package ml

import java.io.{File, PrintWriter}

import org.apache.spark.mllib.tree.model.DecisionTreeModel
import org.apache.spark.{SparkContext, SparkConf}
import org.apache.spark.mllib.util.MLUtils



      //val model = DecisionTree.trainClassifier(trainingData, numClasses, categoricalFeaturesInfo,
      // impurity, maxDepth, maxBins)
      val modelPath = args(2)
      val model = DecisionTreeModel.load(sc, modelPath)


      // Evaluate model on test instances and compute test error
      val labelAndPreds = testData.map { point =>
        val prediction = model.predict(point.features)
        (point.label, prediction)
      };

      val pw = new PrintWriter(new File(args(1)));
      for( a <- labelAndPreds.collect()) {
        pw.write(a._1 + " " + a._2 + "\n")
      }

      val testErr = labelAndPreds.filter(r => r._1 != r._2).count.toDouble / testData.count()
      pw.write("Test Error = " + testErr)
      //println("Learned classification tree model:\n" + model.toDebugString)
      pw.close()
    }catch {
      case e: Exception => println("ERROR. tool unsuccessful:" + e);
    }
    }


  }

开发者ID:HPCL，项目名称:GalacticSpark，代码行数:39，代码来源:DTClassifier.scala

注：本文中的org.apache.spark.mllib.tree.model.DecisionTreeModel类示例整理自Github/MSDocs等源码及文档管理平台，相关代码片段筛选自各路编程大神贡献的开源项目，源码版权归原作者所有，传播和使用请参考对应项目的License；未经允许，请勿转载。