Отображение дерева решений в Спарк MLlib с правильными именами художественных

Question

я построить дерево решений в Спарк MLLib

val dt = new DecisionTreeClassifier().setLabelCol("indexedLabel").setFeaturesCol("indexedFeatures").setImpurity(impurity).setMaxBins(maxBins).setMaxDepth(maxDepth) 

Я показал это следующим образом:

val treeModel = model.stages(1).asInstanceOf[DecisionTreeClassificationModel] 
println("Learned classification tree model:\n" + treeModel.toDebugString) 

В результате текстовое описание дерево с «Features X» вместо фактических имен функций моего набора данных, что делает дерево не интерпретируемым.

У кого-нибудь есть указатели для отображения дерева с правильными именами объектов, позволяющими интерпретировать дерево? Спасибо!!

Answer 1

Модели дерева решений Spark ML (lib) (начиная с версии 1.6.X) не содержат спецификаций функций. В основном, ваш столбец функций («indexedFeatures») представляет собой двойной массив, в котором отдельные элементы массива функционируют как непрерывные пороговые значения разделения.

В соответствующем объекте DataFrame имеется информация о спецификациях функций. Во-первых, проверьте метаданные столбца «indexedFeatures». Если этого недостаточно, проверьте метаданные его составляющих столбцов и т. Д.

Библиотека JPMML-SparkML может использоваться для экспорта моделей трубопроводов Spark ML в формате данных PMML. Его двигатель выполняет отличную коллекцию спецификаций и (реконструкцию) строительных работ.