R randomForest到PMML類索引是錯誤的

Question

我正在將一個R randomForest模型導出到PMML。生成的PMML始終將類作爲DataDictionary元素的第一個元素，但並非總是如此。

有什麼方法可以解決這個問題，或者至少用自定義擴展元素增加PMML？這樣我可以把班級索引放在那裏。

我查看了pmml軟件包文檔以及pmmlTransformations軟件包，但找不到任何可以幫助我解決此問題的內容。

Answer 1

By PMML class我假設你是指PMML模型屬性中的模型類型（分類與迴歸）？ 如果是這樣，模型類型不是從DataDictionary的第一個元素的數據類型確定的...這些是完全獨立的。模型類型由模型類型R認爲它是確定的。 R隨機森林對象確定它認爲它的類型（模型$ type），這是pmml函數導出的模型類型。如果你想讓你的模型成爲某種類型，只要確保讓R知道......例如，如果你使用的是虹膜數據集，如果你的預測變量是Sepal.Length，R就會正確地認爲它是一個迴歸模型。如果您堅持將其作爲分類模型處理，請嘗試使用as.factor（Sepal.Length）。