DecisionTreeClassifier

类 pyspark.ml.classification。 DecisionTreeClassifier ( *,featuresCol:str=“特性”,labelCol:str=“标签”,predictionCol:str=“预测”,probabilityCol:str=“概率”,rawPredictionCol:str=“rawPrediction”,maxDepth:int=5,maxBins:int=32,minInstancesPerNode:int=1,minInfoGain:浮动=0.0,maxMemoryInMB:int=256年,cacheNodeIds:bool=假,checkpointInterval:int=10,杂质:str=“基尼”,种子:可选(int]=没有一个,weightCol:可选(str]=没有一个,leafCol:str=”,minWeightFractionPerNode:浮动=0.0 )

决策树学习算法的分类。它同时支持二进制和多类标签,以及连续和分类功能。

例子

> > >从pyspark.ml.linalg进口向量> > >从pyspark.ml.feature进口StringIndexer> > >df=火花。createDataFrame([…(1.0,向量。密集的(1.0)),…(0.0,向量。稀疏的(1,[],[]))),(“标签”,“特征”])> > >stringIndexer=StringIndexer(inputCol=“标签”,outputCol=“索引”)> > >si_model=stringIndexer。适合(df)> > >道明=si_model。变换(df)> > >dt=DecisionTreeClassifier(maxDepth=2,labelCol=“索引”,leafCol=“leafId”)> > >模型=dt。适合(道明)> > >模型。getLabelCol()“索引”> > >模型。setFeaturesCol(“特征”)DecisionTreeClassificationModel……> > >模型。numNodes3> > >模型。深度1> > >模型。featureImportancesSparseVector ({0: 1.0})> > >模型。numFeatures1> > >模型。numClasses2> > >打印(模型。toDebugString)DecisionTreeClassificationModel…深度= 1,numNodes = 3…> > >电平=火花。createDataFrame(((向量。密集的(- - - - - -1.0),),(“特征”])> > >模型。预测(电平。头()。特性)0.0> > >模型。predictRaw(电平。头()。特性)DenseVector ([1.0, 0.0])> > >模型。predictProbability(电平。头()。特性)DenseVector ([1.0, 0.0])> > >结果=模型。变换(电平)。头()> > >结果。预测0.0> > >结果。概率DenseVector ([1.0, 0.0])> > >结果。rawPredictionDenseVector ([1.0, 0.0])> > >结果。leafId0.0> > >test1=火花。createDataFrame(((向量。稀疏的(1,(0),(1.0),),(“特征”])> > >模型。变换(test1)。头()。预测1.0> > >dtc_path=temp_path+“/ dtc”> > >dt。保存(dtc_path)> > >dt2=DecisionTreeClassifier。负载(dtc_path)> > >dt2。getMaxDepth()2> > >model_path=temp_path+“/ dtc_model”> > >模型。保存(model_path)> > >model2=DecisionTreeClassificationModel。负载(model_path)> > >模型。featureImportances= =model2。featureImportances真正的> > >模型。变换(电平)。取(1)= =model2。变换(电平)。取(1)真正的> > >df3=火花。createDataFrame([…(1.0,0.2,向量。密集的(1.0)),…(1.0,0.8,向量。密集的(1.0)),…(0.0,1.0,向量。稀疏的(1,[],[]))),(“标签”,“重量”,“特征”])> > >si3=StringIndexer(inputCol=“标签”,outputCol=“索引”)> > >si_model3=si3。适合(df3)> > >td3=si_model3。变换(df3)> > >dt3=DecisionTreeClassifier(maxDepth=2,weightCol=“重量”,labelCol=“索引”)> > >model3=dt3。适合(td3)> > >打印(model3。toDebugString)DecisionTreeClassificationModel…深度= 1,numNodes = 3…

方法

清晰的(参数)	清除参数映射的参数是否被显式地设置。
复制((额外的))	创建这个实例的副本具有相同uid和一些额外的参数。
explainParam(参数)	解释一个参数并返回它的名字,医生,和可选的默认值,用户提供的字符串值。
explainParams()	返回文档的所有参数选择默认值和用户提供的值。
extractParamMap((额外的))	提取嵌入默认参数值和用户提供的值,然后合并他们额外的值从输入平面参数映射,后者使用价值如果存在冲突,即。排序:默认参数值< <额外的用户提供的值。
适合(数据集[params))	适合一个模型与可选参数的输入数据集。
fitMultiple(paramMaps数据集)	适合一个模型为每个参数映射的输入数据集paramMaps。
getCacheNodeIds()	得到的价值cacheNodeIds或其默认值。
getCheckpointInterval()	得到的价值checkpointInterval或其默认值。
getFeaturesCol()	得到的价值featuresCol或其默认值。
getImpurity()	杂质的价值或其默认值。
getLabelCol()	得到的价值labelCol或其默认值。
getLeafCol()	得到的价值leafCol或其默认值。
getMaxBins()	得到的价值maxBins或其默认值。
getMaxDepth()	maxDepth的价值或其默认值。
getMaxMemoryInMB()	得到的价值maxMemoryInMB或其默认值。
getMinInfoGain()	得到的价值minInfoGain或其默认值。
getMinInstancesPerNode()	得到的价值minInstancesPerNode或其默认值。
getMinWeightFractionPerNode()	得到的价值minWeightFractionPerNode或其默认值。
getOrDefault(参数)	得到参数的值在用户提供的参数映射或其默认值。
getParam(paramName)	通过它的名称参数。
getPredictionCol()	得到的价值predictionCol或其默认值。
getProbabilityCol()	得到的价值probabilityCol或其默认值。
getRawPredictionCol()	得到的价值rawPredictionCol或其默认值。
getSeed()	种子的价值或其默认值。
getThresholds()	得到的值阈值或其默认值。
getWeightCol()	得到的价值weightCol或其默认值。
hasDefault(参数)	检查是否一个参数有默认值。
hasParam(paramName)	测试这个实例包含一个参数是否与给定名称(字符串)。
isDefined(参数)	检查参数是否由用户或显式地设置一个默认值。
收取(参数)	检查参数是否由用户显式地设置。
负载(路径)	从输入路径,读取一个毫升实例的快捷方式read () .load(路径)。
读()	返回一个MLReader这个类的实例。
保存(路径)	这个毫升实例保存到给定的路径,一个快捷方式的“写().save(路径)。
集(参数值)	设置一个参数嵌入参数映射。
setCacheNodeIds(值)	设置的值cacheNodeIds。
setCheckpointInterval(值)	设置的值checkpointInterval。
setFeaturesCol(值)	设置的值featuresCol。
setImpurity(值)	设置的值杂质。
setLabelCol(值)	设置的值labelCol。
setLeafCol(值)	设置的值leafCol。
setMaxBins(值)	设置的值maxBins。
setMaxDepth(值)	设置的值maxDepth。
setMaxMemoryInMB(值)	设置的值maxMemoryInMB。
setMinInfoGain(值)	设置的值minInfoGain。
setMinInstancesPerNode(值)	设置的值minInstancesPerNode。
setMinWeightFractionPerNode(值)	设置的值minWeightFractionPerNode。
setparam(自我,\ [,labelCol featuresCol…))	DecisionTreeClassifier设置参数。
setPredictionCol(值)	设置的值predictionCol。
setProbabilityCol(值)	设置的值probabilityCol。
setRawPredictionCol(值)	设置的值rawPredictionCol。
setSeed(值)	设置的值种子。
setThresholds(值)	设置的值阈值。
setWeightCol(值)	设置的值weightCol。
写()	返回一个MLWriter实例毫升实例。

属性

cacheNodeIds
checkpointInterval
featuresCol
杂质
labelCol
leafCol
maxBins
maxDepth
maxMemoryInMB
minInfoGain
minInstancesPerNode
minWeightFractionPerNode
参数个数	返回所有参数命令的名字。
predictionCol
probabilityCol
rawPredictionCol
种子
supportedImpurities
阈值
weightCol

方法的文档

清晰的 ( 参数:pyspark.ml.param.Param )→没有

清除参数映射的参数是否被显式地设置。

复制 ( 额外的:可选(ParamMap]=没有一个 )→摩根大通

创建这个实例的副本具有相同uid和一些额外的参数。这个实现第一次调用参数。复制and then make a copy of the companion Java pipeline component with extra params. So both the Python wrapper and the Java pipeline component get copied.

参数

额外的 东西,可选

额外参数复制到新实例

返回

JavaParams

这个实例的副本

explainParam ( 参数:联盟(str,pyspark.ml.param.Param] )→str

解释一个参数并返回它的名字,医生,和可选的默认值,用户提供的字符串值。

explainParams ( )→str

返回文档的所有参数选择默认值和用户提供的值。

extractParamMap ( 额外的:可选(ParamMap]=没有一个 )→ParamMap

提取嵌入默认参数值和用户提供的值,然后合并他们额外的值从输入平面参数映射,后者使用价值如果存在冲突,即。排序:默认参数值< <额外的用户提供的值。

参数

额外的 东西,可选

额外的参数值

返回

dict

合并后的参数映射

适合 ( 数据集:pyspark.sql.dataframe.DataFrame,参数个数:联盟[ParamMap、列表(ParamMap)元组(ParamMap),没有一个)=没有一个 )→联盟(米,列表(米] ]

适合一个模型与可选参数的输入数据集。

参数

数据集 pyspark.sql.DataFrame

输入数据集。

参数个数 dict或列表或元组,可选的

一个可选的参数覆盖嵌入参数的地图。如果列表/元组的参数映射,这适用于每一个参数并返回一个列表的模型映射。

返回

变压器或者一个列表变压器

拟合模型(年代)

fitMultiple ( 数据集:pyspark.sql.dataframe.DataFrame,paramMaps:序列(ParamMap] )→迭代器(元组(int,米] ]

适合一个模型为每个参数映射的输入数据集paramMaps。

参数

数据集 pyspark.sql.DataFrame

输入数据集。

paramMaps collections.abc.Sequence

一系列的参数映射。

返回

_FitMultipleIterator

一个线程安全的iterable包含为每个参数映射模型。每次调用下一个(modelIterator)将返回(指数模型)模型是适合使用在哪里paramMaps(指数)。指数值可能不是连续的。

getCacheNodeIds ( )→bool

得到的价值cacheNodeIds或其默认值。

getCheckpointInterval ( )→int

得到的价值checkpointInterval或其默认值。

getFeaturesCol ( )→str

得到的价值featuresCol或其默认值。

getImpurity ( )→str

杂质的价值或其默认值。

getLabelCol ( )→str

得到的价值labelCol或其默认值。

getLeafCol ( )→str

得到的价值leafCol或其默认值。

getMaxBins ( )→int

得到的价值maxBins或其默认值。

getMaxDepth ( )→int

maxDepth的价值或其默认值。

getMaxMemoryInMB ( )→int

得到的价值maxMemoryInMB或其默认值。

getMinInfoGain ( )→浮动

得到的价值minInfoGain或其默认值。

getMinInstancesPerNode ( )→int

得到的价值minInstancesPerNode或其默认值。

getMinWeightFractionPerNode ( )→浮动

得到的价值minWeightFractionPerNode或其默认值。

getOrDefault ( 参数:联盟(str,pyspark.ml.param.Param(T]] )→联盟(任何,T]

得到参数的值在用户提供的参数映射或其默认值。如果没有设置提出了一个错误。

getParam ( paramName:str )→pyspark.ml.param.Param

通过它的名称参数。

getPredictionCol ( )→str

得到的价值predictionCol或其默认值。

getProbabilityCol ( )→str

得到的价值probabilityCol或其默认值。

getRawPredictionCol ( )→str

得到的价值rawPredictionCol或其默认值。

getSeed ( )→int

种子的价值或其默认值。

getThresholds ( )→列表(浮动]

得到的值阈值或其默认值。

getWeightCol ( )→str

得到的价值weightCol或其默认值。

hasDefault ( 参数:联盟(str,pyspark.ml.param.Param(任何]] )→bool

检查是否一个参数有默认值。

hasParam ( paramName:str )→bool

测试这个实例包含一个参数是否与给定名称(字符串)。

isDefined ( 参数:联盟(str,pyspark.ml.param.Param(任何]] )→bool

检查参数是否由用户或显式地设置一个默认值。

收取 ( 参数:联盟(str,pyspark.ml.param.Param(任何]] )→bool

检查参数是否由用户显式地设置。

classmethod 负载 ( 路径:str )→RL

从输入路径,读取一个毫升实例的快捷方式read () .load(路径)。

classmethod 读 ( )→pyspark.ml.util.JavaMLReader(RL]

返回一个MLReader这个类的实例。

保存 ( 路径:str )→没有

这个毫升实例保存到给定的路径,一个快捷方式的“写().save(路径)。

集 ( 参数:pyspark.ml.param.Param,价值:任何 )→没有

设置一个参数嵌入参数映射。

setCacheNodeIds ( 价值:bool )→pyspark.ml.classification.DecisionTreeClassifier

设置的值cacheNodeIds。

setCheckpointInterval ( 价值:int )→pyspark.ml.classification.DecisionTreeClassifier

设置的值checkpointInterval。

setFeaturesCol ( 价值:str )→P

设置的值featuresCol。

setImpurity ( 价值:str )→pyspark.ml.classification.DecisionTreeClassifier

设置的值杂质。

setLabelCol ( 价值:str )→P

设置的值labelCol。

setLeafCol ( 价值:str )→P

设置的值leafCol。

setMaxBins ( 价值:int )→pyspark.ml.classification.DecisionTreeClassifier

设置的值maxBins。

setMaxDepth ( 价值:int )→pyspark.ml.classification.DecisionTreeClassifier

设置的值maxDepth。

setMaxMemoryInMB ( 价值:int )→pyspark.ml.classification.DecisionTreeClassifier

设置的值maxMemoryInMB。

setMinInfoGain ( 价值:浮动 )→pyspark.ml.classification.DecisionTreeClassifier

设置的值minInfoGain。

setMinInstancesPerNode ( 价值:int )→pyspark.ml.classification.DecisionTreeClassifier

设置的值minInstancesPerNode。

setMinWeightFractionPerNode ( 价值:浮动 )→pyspark.ml.classification.DecisionTreeClassifier

设置的值minWeightFractionPerNode。

setparam ( 自我,\ *,featuresCol = "特性",labelCol = "标签",predictionCol = "预测",probabilityCol =“概率”,rawPredictionCol = " rawPrediction ",maxDepth = 5,maxBins = 32,minInstancesPerNode = 1,minInfoGain = 0.0,maxMemoryInMB = 256,cacheNodeIds = False,checkpointInterval = 10,杂质=“基尼”,种子=没有,weightCol =没有,leafCol = " ",minWeightFractionPerNode = 0.0 )

DecisionTreeClassifier设置参数。

setPredictionCol ( 价值:str )→P

设置的值predictionCol。

setProbabilityCol ( 价值:str )→P

设置的值probabilityCol。

setRawPredictionCol ( 价值:str )→P

设置的值rawPredictionCol。

setSeed ( 价值:int )→pyspark.ml.classification.DecisionTreeClassifier

设置的值种子。

setThresholds ( 价值:列表(浮动] )→P

设置的值阈值。

setWeightCol ( 价值:str )→pyspark.ml.classification.DecisionTreeClassifier

设置的值weightCol。

写 ( )→pyspark.ml.util.JavaMLWriter

返回一个MLWriter实例毫升实例。

属性的文档

cacheNodeIds =参数(父母=‘定义’,name = ' cacheNodeIds ', doc = '如果错误,算法将树木与节点执行人匹配实例。如果这是真的,该算法将缓存节点为每个实例id。缓存可以加快训练更深层次的树。用户可以设置缓存应该多久通过设置checkpointInterval检查点或禁用它。”)

checkpointInterval =参数(父母=‘定义’,name = ' checkpointInterval ', doc = '设置检查点间隔(> = 1)或禁用检查点(1)。例如10意味着缓存将检查点每10迭代。注意:此设置将被忽略,如果目录没有设置检查站SparkContext。”)

featuresCol =参数(父母=‘定义’,name = ' featuresCol ', doc =功能列名称。)

杂质 =参数(父母=‘定义’,name =“杂质”,医生= '标准用于信息增益计算(不区分大小写)。支持选择:熵,基尼”)

labelCol =参数(父母=‘定义’,name = ' labelCol ', doc =“标签列名。”)

leafCol =参数(父母=‘定义’,name = ' leafCol ', doc = '叶指数列名。预测叶指数每棵树的每个实例预订。)

maxBins =参数(父母=‘定义’,name = ' maxBins ', doc = '最大数量的垃圾箱离散化连续特性。必须> = 2,> =数量的类别分类特性。”)

maxDepth =参数(父母=‘定义’,name = ' maxDepth ',医生= '树的最大深度。(> = 0)例如,深度0意味着1叶节点;深度1意味着1 + 2叶节点内部节点。必须在区间[0,30]。”)

maxMemoryInMB =参数(父母=‘定义’,name = ' maxMemoryInMB ',医生在MB = '最大内存分配给直方图聚合。如果太小,那么1个节点将被分配每个迭代,和它的总量可能超过这个大小。”)

minInfoGain =参数(父母=‘定义’,name = ' minInfoGain ', doc =分割的最小信息增益被认为是在一个树节点。)

minInstancesPerNode =参数(父母=‘定义’,name = ' minInstancesPerNode ', doc = '每个孩子都必须有最小数量的实例后分裂。如果分裂导致左边或者右边的孩子不到minInstancesPerNode,分割将作为无效的被丢弃。应该是> = 1”。)

minWeightFractionPerNode =参数(父母=‘定义’,name = ' minWeightFractionPerNode ', doc = '的最低分数加权样本计数后,每个孩子都必须有分裂。如果一个分裂导致分数总重量的向左或向右孩子小于minWeightFractionPerNode,分割将作为无效的被丢弃。应该在区间[0.0,0.5)。”)

参数个数

返回所有参数命令的名字。默认实现使用dir ()所有的属性类型参数。

predictionCol =参数(父母=‘定义’,name = ' predictionCol ', doc =预测列名称。)

probabilityCol =参数(父母=‘定义’,name = ' probabilityCol ', doc = '列名为预测类条件概率。注意:并不是所有的模型输出精确校准的概率估计!这些概率应该被视为机密,而不是精确的概率。”)

rawPredictionCol =参数(父母=‘定义’,name = ' rawPredictionCol ', doc =“原始预测(又名信心)列名”。)

种子 =参数(父母=‘定义’,name =“种子”,医生=“随机种子。”)

supportedImpurities =(“熵”、“基尼”)

阈值 =参数(父母=‘定义’,name =“阈值”,医生=“多层次分类阈值调整的概率预测每个类。数组长度必须等于类的数量,最多值> 0,除了一个值可能是0。类最大的值p / t是预测,p是原始类和t的概率是类的门槛。”)

weightCol =参数(父母=‘定义’,name = ' weightCol ', doc = '体重列名。如果这不是设置或空,我们对所有实例权重为1.0。”)

以前的

BinaryLogisticRegressionTrainingSummary

下一个

DecisionTreeClassificationModel