optimize E

Author: WenDesi

.idea/workspace.xml

@@ -5,6 +5,7 @@
 # @Last modified by:   WenDesi
 # @Last modified time: 12-11-16
 
+import time
 import csv
 import random
 
@@ -216,9 +217,12 @@ def split_features_labels(dataset):
     # train_features,train_labels = [[3,3],[4,4],[1,1]],[1,1,-1]
     # test_features,test_labels = [[0,0],[0,3],[0,6]],[-1,-1,1]
 
+    time1 = time.time()
     svm = SVM()
     svm.train(train_features,train_labels)
     test_predict = svm.predict(test_features)
 
     accuracy = sum([test_labels[i]==test_predict[i] for i in xrange(len(test_predict))])
     print 'accuracy is ', float(accuracy)/float(len(test_labels))
+    time2 = time.time()
+    print 'cost ',time2-time1