pyspark --- randomForests使用「categoricalFeaturesInfo」指定分類變量

Question

如何在pyspark randomForests中指定categoricalFeaturesInfo？

文件是不是在該很清楚，我嘗試了一些這樣的：

categoricalFeaturesInfo = {（12,4）}

categoricalFeaturesInfo = {（12-> 4）}

categoricalFeaturesInfo = {Map [int，int]（12,4）}

...等等，它們都不起作用。任何幫助是極大的讚賞。

Answer 1

從Python文檔我們有：

categoricalFeaturesInfo: Map storing arity of categorical 
      features. E.g., an entry (n -> k) indicates that 
      feature n is categorical with k categories indexed 
      from 0: {0, 1, ..., k-1}. 

嘗試使用：

categoricalFeaturesInfo = {12:4} 

Answer 2

對於火花2.0及以上

它的一個字典 與密鑰作爲變量的索引/位置rdd（功能） 和值作爲類別的數量