Text-Classification-544/main.py at master · ChenX1993/Text-Classification-544 · GitHub

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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
# -- coding: utf-8 --
import os
import time

def CNN():
	startTime = time.time()
	print ''
	print '------CNN------'
	print ''
	os.system('python cnn/cnn.py')
	print ''
	print '-Step 4: test data...'
	os.system('python cnn/prediction.py')

	duringTime = time.time() - startTime
	print 'Duration: %.3f s' %duringTime
	print '------CNN complete------'

def KNN():
	startTime = time.time()
	print ''
	print '------KNN------'
	print ''

	print'-Step 1: feature selection...'
	os.system('python knn/my_feature_selection.py')

	print '-Step 2: calculate training feature weight...'
	os.system('python knn/my_feature_weight.py')

	print '-Step 3: calculate test feature weight...'
	os.system('python knn/my_test_feature_weight.py')

	print '-Step 4: train model and test data...'
	os.system('python knn/KNN.py')

	duringTime = time.time() - startTime
	print 'Duration: %.3f s' %duringTime
	print '------KNN complete------'

def SVM():
	startTime = time.time()
	print ''
	print '------SVM------'
	print ''

	print'-Step 1: feature selection...'
	os.system('python svm/feature_selection.py')

	print '-Step 2: calculate training feature weight...'
	os.system('python svm/feature_weight.py')

	print '-Step 3: calculate test feature weight...'
	os.system('python svm/test_feature_weight.py')

	print '-Step 4: dimensionality reduction...'
	os.system('python svm/ReduceDimension.py')

	print '-Step 5: train model and test data...'
	os.system('python svm/testsvm.py')

	duringTime = time.time() - startTime
	print 'Duration: %.3f s' %duringTime
	print '------SVM complete------'


def NB():
	startTime = time.time()
	print ''
	print '------Naiev Bayes------'
	print ''

	print '-Step 1: model training...'
	os.system('python nb/nblearn.py')

	print '-Step 2: test data...'
	os.system('python nb/nbclassify.py')

	duringTime = time.time() - startTime
	print 'Duration: %.3f s' %duringTime
	print '------Naiev Bayes complete------'


def Results():
	print ''
	print '------Show Results------'
	print ''
	os.system('python combine_result.py')
	print ''
	print 'Prediction results saved in : ./rusult/all_result.csv'
	print ''
	print '------Show Results complete------'
print '------Text Classification------'

while True:
	print ''
	print ''
	print '[1] CNN'
	print '[2] KNN'
	print '[3] SVM'
	print '[4] Naive Bayes'
	print '[5] Show Current Results'
	print '[6] quit'
	print ''

	inputStr = raw_input('Choose your classification method (only number valid): ')

	if inputStr == '1':
		CNN()
	elif inputStr == '2':
		KNN()
	elif inputStr == '3':
		SVM()
	elif inputStr == '4':
		NB()
	elif inputStr == '5':
		Results()
	elif inputStr == '6':
		break
	else:
		print '*Warning! Your input is invalid. Please enter a correct input.'
		continue