Skip to content

Commit de0dedb

Browse files
Update random_forests_train.py
1 parent 078c622 commit de0dedb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Chapter_5 Random Forest/random_forests_train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,8 @@ def choose_samples(data, k):
4747
for i in xrange(m):
4848
data_tmp = []
4949
for fea in feature:
50-
data_tmp.append(data[i][fea])
51-
data_tmp.append(data[i][-1])
50+
data_tmp.append(data[index[i]][fea])
51+
data_tmp.append(data[index[i]][-1])
5252
data_samples.append(data_tmp)
5353
return data_samples, feature
5454

0 commit comments

Comments
 (0)