首页 > spark mllib里决策树的设置参数里categoricalFeaturesInfo是什么意思?

spark mllib里决策树的设置参数里categoricalFeaturesInfo是什么意思?

看官方文档的意思是将类型特征的索引映射到特征中类型的数目,但还是不明白,比如我有一个类型特征feat=["a","b","c","d"],需要将a,b,c,d的索引都要映射一下吗?


不用,那个对象是用来声明特征值的,
Map<Integer, Integer> categoricalFeaturesInfo = new HashMap<Integer, Integer>();
只要定义它出来即可

【热门文章】
【热门文章】