在RandomForestClassifier中使用复值数据

时间:2015-11-02 17:23:18

标签: python numpy scikit-learn random-forest complex-numbers

我有一个频域值的数据集,所以每个元素的末尾都有一个“j”,如“27.6695554346-5.51875942715j”。&#xA ;这是矩阵中某些数据的图片









当我尝试将 RandomForestClassifier 与以下代码一起使用时:




  cfr = RandomForestClassifier(n_estimators = 10,n_jobs = 5 )
 cfr.fit(X1,y1)
 predictY2 = cfr.predict(X2)
  




我有此错误:




  ValueError:无法将字符串转换为float:(27.6695554346-5.51875942715j)
  
&# xA;


请您解释一下我要做些什么来解决问题?




1 个答案:

答案 0 :(得分:0)

必须将它们转换为绝对值 所以只要在矩阵上应用np.abs,问题就解决了!