apache-spark - pyspark---randomForests 使用 "categoricalFeaturesInfo"指定分类变量

Question

如何在 pyspark randomForests 中指定 categoricalFeaturesInfo？

文档对此不是很清楚，我尝试了一些类似的:

分类特征信息= {(12,4)}

分类特征信息= {(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}