Spark 2.x 决策树示例代码-IRIS数据集

最新推荐文章于 2024-12-07 10:51:27 发布

谁用了尧哥这个昵称

最新推荐文章于 2024-12-07 10:51:27 发布

阅读量962

点赞数

CC 4.0 BY-SA版权

分类专栏： AI 文章标签： spark2.X 决策树

本文链接：https://blog.youkuaiyun.com/xuyaoqiaoyaoge/article/details/80024038

本文通过Spark 2.x详细介绍如何使用决策树进行 IRIS 数据集的分类，首先提供了数据集下载链接，然后详细展示并解释了核心代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

数据集下载

下载链接

代码

package Iris;

import org.apache.spark.api.java.JavaPairRDD;
import org.apache.spark.api.java.JavaRDD;
import org.apache.spark.ml.Pipeline;
import org.apache.spark.ml.PipelineModel;
import org.apache.spark.ml.PipelineStage;
import org.apache.spark.ml.Transformer;
import org.apache.spark.ml.classification.DecisionTreeClassificationModel;
import org.apache.spark.ml.classification.DecisionTreeClassifier;
import org.apache.spark.ml.evaluation.MulticlassClassificationEvaluator;
import org.apache.spark.ml.feature.*;
import org.apache.spark.ml.linalg.VectorUDT;
import org.apache.spark.ml.linalg.Vectors;
import org.apache.spark.sql.Dataset;
import org.apache.spark.sql.Row;
import org