test_model.py
968 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
__author__ = 'chunk'
from common import *
from mdata import MSR, CV
from mmodel import SVM
from mfeat import HOG
timer = Timer()
def test_SVM_CV():
timer.mark()
dcv = CV.DataCV()
X, Y = dcv.load_data(mode='local') # 90.468586s -> 5.392520s
# X, Y = dcv.load_data(mode='hbase') # 21.682754s
# X, Y = dcv.load_data(mode='spark') # 29.549597s
timer.report()
timer.mark()
msvm = SVM.ModelSVM()
msvm.train(X, Y)
timer.report()
timer.mark()
for path, subdirs, files in os.walk('data/467/'):
for name in files:
imgpath = os.path.join(path, name)
feat = dcv.get_feat(imgpath, 'hog')
print name, msvm.predict(feat)
timer.report()
timer.mark()
print msvm.test(X, Y) # 0.948892561983 for svm_cv, 0.989024793388 for svm_sk, 0.9900826446280992 for svm_lib
timer.report() # 27.421949s for svm_lib
if __name__ == '__main__':
test_SVM_CV()
print 'helllo'