【学习笔记】pyspark-决策树～ from pyspark.ml.feature import VectorAssemb ...

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

pyspark-决策树～
from pyspark.ml.feature import VectorAssembler
生成的spark_dataframe需要先进行vectorassembler进行转换，才能调用spark的机器学习算法～

.randomSplit
切分数据集，只能将数据切分为训练机和测试集，xy不分

from pyspark.ml.classification import DecisionTreeClassifier
导入分类决策树算法

from pyspark.ml.evaluation import MulticlassClassificationEvaluator
模型评估的库～但是召回率计算规则不明，需要自己计算～

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：feature Vector import Spark SPAR

赞