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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
#!/usr/bin/python3
import os
import os.path
import shutil
import sys
from subprocess import Popen, PIPE
from argparse import ArgumentParser
from myconfig import MyConfig
config = MyConfig()
#change cwd to the libpinyin utils/training directory
libpinyin_dir = config.getToolsDir()
libpinyin_sub_dir = os.path.join(libpinyin_dir, 'utils', 'training')
os.chdir(libpinyin_sub_dir)
#chdir done
def validateModel(modelfile):
#begin processing
cmdline = ['./validate_k_mixture_model', \
modelfile]
subprocess = Popen(cmdline, shell=False, close_fds=True)
#check os.waitpid doc
(pid, status) = os.waitpid(subprocess.pid, 0)
if status != 0:
sys.exit('Corrupted model found when validating:' + modelfile)
#end processing
def exportModel(modelfile, textmodel):
#begin processing
cmdline = ['./export_k_mixture_model', \
'--k-mixture-model-file', \
modelfile]
subprocess = Popen(cmdline, shell=False, stdout=PIPE, \
close_fds=True)
with open(textmodel, 'wb') as f:
f.writelines(subprocess.stdout.readlines())
f.close()
#check os.waitpid doc
(pid, status) = os.waitpid(subprocess.pid, 0)
if status != 0:
sys.exit('Corrupted model found when exporting:' + modelfile)
#end processing
def mergeOneModel(mergedmodel, onemodel, score):
#validate first
validateModel(onemodel)
onemodelstatuspath = onemodel + config.getStatusPostfix()
onemodelstatus = utils.load_status(onemodelstatuspath)
if not utils.check_epoch(onemodelstatus, 'Estimate'):
raise utils.Epoch('Please estimate first.\n')
if score != onemodelstatus['EstimateScore']:
raise AssertionError('estimate scores mis-match.\n')
#begin processing
cmdline = ['./merge_k_mixture_model', \
'--result-file', \
mergedmodel, \
onemodel]
subprocess = Popen(cmdline, shell=False, close_fds=True)
#check os.waitpid doc
(pid, status) = os.waitpid(subprocess.pid, 0)
if status != 0:
sys.exit('Corrupted model found when merging:' + onemodel)
#end processing
def mergeSomeModels(tryname, mergedmodel, sortedindexname, mergenum):
last_score = 1.
#begin processing
indexfile = open(sortedindexname, 'r')
for i in range(mergenum):
line = indexfile.readline()
if not line:
raise AssertionError('No more models.\n')
line = line.rstrip(os.linesep)
(subdir, modelname, score) = line.split('#', 2)
score = float(score)
if score > last_score:
raise AssertionError('score must be descending.\n')
onemodel = os.path.join(config.getModelDir(), subdir, modelname)
mergeOneModel(mergedmodel, onemodel, score)
last_score = score
indexfile.close()
#end processing
#validate merged model
validateModel(mergedmodel)
def pruneModel(modelfile, k, CDF):
#begin processing
cmdline = ['./prune_k_mixture_model', \
'-k', k, '--CDF', CDF,
modelfile]
subprocess = Popen(cmdline, shell=False, close_fds=True)
#check os.waitpid doc
(pid, status) = os.waitpid(subprocess.pid, 0)
if (status != 0):
sys.exit('Corrupted model found when pruning:' + modelfile)
#end processing
if __name__ == '__main__':
parser = ArgumentParser(description='Try prune models.')
parser.add_argument('--modeldir', action='store', \
help='model directory', \
default=config.getModelDir())
parser.add_argument('--mergenumber', action='store', \
help='number of documents to be merged', \
default=10, type=int)
parser.add_argument('-k', action='store', \
help='k parameter of k mixture model prune', \
default=3, type=int)
parser.add_argument('--CDF', action='store', \
help='CDF parameter of k mixture model prune', \
default=0.99, type=float)
parser.add_argument('tryname', action='store', \
help='the storage directory')
args = parser.parse_args()
print(args)
tryname = 'try' + args.tryname
#merge model candidates
mergedmodel = os.path.join(config.getFinalDir(), tryname, 'merged.db')
sortedindexname = os.path.join(args.modeldir, \
config.getSortedEstimateIndex())
mergeSomeModels(tryname, mergedmodel, sortedindexname, args.mergenumber)
#export textual format
exportfile = os.path.join(config.getFinalDir(), tryname, 'kmm_merged.text')
exportModel(mergedmodel, exportfile)
#prune merged model
prunedmodel = os.path.join(config.getFinalDir(), tryname, 'pruned.db')
#backup merged model
shutil.copyfile(mergedmodel, prunedmodel)
pruneModel(prunedmodel, args.k, args.CDF)
#export textual format
exportfile = os.path.join(config.getFinalDir(), tryname, 'kmm_pruned.text')
exportModel(prunedmodel, exportModel)
|