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

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

阅读 4.9k
1 个回答

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

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进