Skip to content

Commit 73f3c9b

Browse files
committed
add code for shuffle data
1 parent 5e2e8c8 commit 73f3c9b

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

DeepLearning Tutorials/dive_into_keras/cnn-svm.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from sklearn.ensemble import RandomForestClassifier
1111
from sklearn.svm import SVC
1212
from data import load_data
13+
import random
1314

1415

1516
def svc(traindata,trainlabel,testdata,testlabel):
@@ -33,8 +34,14 @@ def rf(traindata,trainlabel,testdata,testlabel):
3334
print("cnn-rf Accuracy:",accuracy)
3435

3536
if __name__ == "__main__":
36-
#load data,split into traindata and testdata
37+
#load data
3738
data, label = load_data()
39+
#shuffle the data
40+
index = [i for i in range(len(data))]
41+
random.shuffle(index)
42+
data = data[index]
43+
label = label[index]
44+
3845
(traindata,testdata) = (data[0:30000],data[30000:])
3946
(trainlabel,testlabel) = (label[0:30000],label[30000:])
4047
#use origin_model to predict testdata

0 commit comments

Comments
 (0)