回归模型的特征重要性(DecisionTreeRegressor,IsotonicRegression,GBTRegressor)

时间:2017-04-15 02:58:38

标签: apache-spark-mllib

在我的spark ml应用程序中。我正在使用几个回归模型,如DecisionTreeRegressorModel,LinearRegressionModel,RandomForestRegressorModel,GBTRegressorModel,AFTSurvivalRegressionModel,IsotonicRegressionModel。

我需要在训练样本tarning集时获得所有这些模型的特征权重。在Spark MLib 1.6.2的文档中看来,似乎只有LinearREgressionModel,AFTSurvivalRegression和RandomForestRegressorModel给我一些参数来提取特征权重。线性回归模型和AFTSurvivalRegression为我提供了特征"系数"随机森林给了我"特征重要性"。

其他模型如DecisionTreeRegressor,GBTRegressor,IsotonicRegression是什么?我是否可以在训练样本上从这些模型中获取任何特征重要性参数?

0 个答案:

没有答案