Update random_forests_train.py

zhaozhiyong19890102 · web-flow · commit de0dedbc341b · 2017-08-04T14:50:06.000+08:00
diff --git a/Chapter_5 Random Forest/random_forests_train.py b/Chapter_5 Random Forest/random_forests_train.py
@@ -47,8 +47,8 @@ def choose_samples(data, k):
 for i in xrange(m):
 data_tmp = []
 for fea in feature:
- data_tmp.append(data[i][fea])
- data_tmp.append(data[i][-1])
+ data_tmp.append(data[index[i]][fea])
+ data_tmp.append(data[index[i]][-1])
 data_samples.append(data_tmp)
 return data_samples, feature
 

-Original file line number
+Diff line change
 foriinxrange(m):
 data_tmp= []
 forfeainfeature:
 -data_tmp.append(data[i][fea])
 -data_tmp.append(data[i][-1])
 +data_tmp.append(data[index[i]][fea])
 +data_tmp.append(data[index[i]][-1])
 data_samples.append(data_tmp)
 returndata_samples, feature