RandomForestRegressor

类 pyspark.ml.regression。 RandomForestRegressor ( *,featuresCol:str=“特性”,labelCol:str=“标签”,predictionCol:str=“预测”,maxDepth:int=5,maxBins:int=32,minInstancesPerNode:int=1,minInfoGain:浮动=0.0,maxMemoryInMB:int=256年,cacheNodeIds:bool=假,checkpointInterval:int=10,杂质:str=“方差”,subsamplingRate:浮动=1.0,种子:可选(int]=没有一个,numTrees:int=20.,featureSubsetStrategy:str=“汽车”,leafCol:str=”,minWeightFractionPerNode:浮动=0.0,weightCol:可选(str]=没有一个,引导:可选(bool]=真正的 )

随机森林为回归学习算法。它同时支持连续和分类功能。

例子

> > >从numpy进口allclose> > >从pyspark.ml.linalg进口向量> > >df=火花。createDataFrame([…(1.0,向量。密集的(1.0)),…(0.0,向量。稀疏的(1,[],[]))),(“标签”,“特征”])> > >射频=RandomForestRegressor(numTrees=2,maxDepth=2)> > >射频。getMinWeightFractionPerNode()0.0> > >射频。setSeed(42)RandomForestRegressor……> > >模型=射频。适合(df)> > >模型。getBootstrap()真正的> > >模型。getSeed()42> > >模型。setLeafCol(“leafId”)RandomForestRegressionModel……> > >模型。featureImportancesSparseVector ({0: 1.0})> > >allclose(模型。treeWeights,(1.0,1.0])真正的> > >电平=火花。createDataFrame(((向量。密集的(- - - - - -1.0),),(“特征”])> > >模型。预测(电平。头()。特性)0.0> > >模型。predictLeaf(电平。头()。特性)DenseVector ([0.0, 0.0])> > >结果=模型。变换(电平)。头()> > >结果。预测0.0> > >结果。leafIdDenseVector ([0.0, 0.0])> > >模型。numFeatures1> > >模型。树(DecisionTreeRegressionModel…深度=…,DecisionTreeRegressionModel……)> > >模型。getNumTrees2> > >test1=火花。createDataFrame(((向量。稀疏的(1,(0),(1.0),),(“特征”])> > >模型。变换(test1)。头()。预测0.5> > >rfr_path=temp_path+“/ rfr”> > >射频。保存(rfr_path)> > >rf2=RandomForestRegressor。负载(rfr_path)> > >rf2。getNumTrees()2> > >model_path=temp_path+“/ rfr_model”> > >模型。保存(model_path)> > >model2=RandomForestRegressionModel。负载(model_path)> > >模型。featureImportances= =model2。featureImportances真正的> > >模型。变换(电平)。取(1)= =model2。变换(电平)。取(1)真正的

方法

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

属性

引导
cacheNodeIds
checkpointInterval
featureSubsetStrategy
featuresCol
杂质
labelCol
leafCol
maxBins
maxDepth
maxMemoryInMB
minInfoGain
minInstancesPerNode
minWeightFractionPerNode
numTrees
参数个数	返回所有参数命令的名字。
predictionCol
种子
subsamplingRate
supportedFeatureSubsetStrategies
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(指数)。指数值可能不是连续的。

getBootstrap ( )→bool

被引导的价值或其默认值。

getCacheNodeIds ( )→bool

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

getCheckpointInterval ( )→int

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

getFeatureSubsetStrategy ( )→str

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

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或其默认值。

getNumTrees ( )→int

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

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

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

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

通过它的名称参数。

getPredictionCol ( )→str

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

getSeed ( )→int

种子的价值或其默认值。

getSubsamplingRate ( )→浮动

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

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,价值:任何 )→没有

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

setBootstrap ( 价值:bool )→pyspark.ml.regression.RandomForestRegressor

设置的值引导。

setCacheNodeIds ( 价值:bool )→pyspark.ml.regression.RandomForestRegressor

设置的值cacheNodeIds。

setCheckpointInterval ( 价值:int )→pyspark.ml.regression.RandomForestRegressor

设置的值checkpointInterval。

setFeatureSubsetStrategy ( 价值:str )→pyspark.ml.regression.RandomForestRegressor

设置的值featureSubsetStrategy。

setFeaturesCol ( 价值:str )→P

设置的值featuresCol。

setImpurity ( 价值:str )→pyspark.ml.regression.RandomForestRegressor

设置的值杂质。

setLabelCol ( 价值:str )→P

设置的值labelCol。

setLeafCol ( 价值:str )→P

设置的值leafCol。

setMaxBins ( 价值:int )→pyspark.ml.regression.RandomForestRegressor

设置的值maxBins。

setMaxDepth ( 价值:int )→pyspark.ml.regression.RandomForestRegressor

设置的值maxDepth。

setMaxMemoryInMB ( 价值:int )→pyspark.ml.regression.RandomForestRegressor

设置的值maxMemoryInMB。

setMinInfoGain ( 价值:浮动 )→pyspark.ml.regression.RandomForestRegressor

设置的值minInfoGain。

setMinInstancesPerNode ( 价值:int )→pyspark.ml.regression.RandomForestRegressor

设置的值minInstancesPerNode。

setMinWeightFractionPerNode ( 价值:浮动 )→pyspark.ml.regression.RandomForestRegressor

设置的值minWeightFractionPerNode。

setNumTrees ( 价值:int )→pyspark.ml.regression.RandomForestRegressor

设置的值numTrees。

setparam ( 自我,\ *,featuresCol = "特性",labelCol = "标签",predictionCol = "预测",maxDepth = 5,maxBins = 32,minInstancesPerNode = 1,minInfoGain = 0.0,maxMemoryInMB = 256,cacheNodeIds = False,checkpointInterval = 10,杂质=“方差”,subsamplingRate = 1.0,种子=没有,numTrees = 20,featureSubsetStrategy =“自动”,leafCol = " ",minWeightFractionPerNode = 0.0,weightCol =没有,引导= True )

设置参数的线性回归。

setPredictionCol ( 价值:str )→P

设置的值predictionCol。

setSeed ( 价值:int )→pyspark.ml.regression.RandomForestRegressor

设置的值种子。

setSubsamplingRate ( 价值:浮动 )→pyspark.ml.regression.RandomForestRegressor

设置的值subsamplingRate。

setWeightCol ( 价值:str )→pyspark.ml.regression.RandomForestRegressor

设置的值weightCol。

写 ( )→pyspark.ml.util.JavaMLWriter

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

属性的文档

引导 =参数(父母=‘定义’,name =“引导”,医生=“引导样品是否使用在构建树”。)

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

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

featureSubsetStrategy =参数(父母=‘定义’,name = ' featureSubsetStrategy ', doc = "的特性考虑数量在每个树节点分裂。支持选择:“汽车”(自动选择任务:如果numTrees = = 1,设置为“所有”。如果numTrees > 1(森林),设置为“√”为分类和回归“onethird”),“所有”(使用所有功能),“onethird”(使用1/3的特性),“√”(调用sqrt(特性)的数量),“log2”(用log2(特性)的数量),“n”(当n的范围(0,1.0],使用n *数量的特性。当n的范围(1,数量的功能),使用n特性)。默认=汽车”)

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)。”)

numTrees =参数(父母=‘定义’,name = ' numTrees ', doc =数量的树木训练(> = 1)。)

参数个数

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

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

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

subsamplingRate =参数(父母=‘定义’,name = ' subsamplingRate ', doc = '训练数据的一部分用于学习每一个决策树,在范围(0,1]。)

supportedFeatureSubsetStrategies =[‘汽车’,‘所有’,‘onethird’,‘√’,‘log2’)

supportedImpurities =(“方差”)

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

以前的

LinearRegressionTrainingSummary

下一个

RandomForestRegressionModel