def viterbi():
observationProb, keyOrder, length = hmmIO.readFile(
'../data/rnnTest.prb')
outputData = {}
trainLabel = hmmIO.readTrainLabel('../data/label/train.lab')
transMatrix = hmmFunc.transProb(trainLabel)
def forward(key, preValue, preTrace):
reValue = []
reTrace = []
for i in range(0, 48):
tempValue = [
preValue[prei] + math.log(transMatrix[prei][i]) +
math.log(observationProb[key][i]) for prei in range(0, 48)
]
maxIndex = np.argmax(tempValue)
reValue.append(tempValue[maxIndex])
tempTrace = preTrace[maxIndex][:]
reTrace.append(tempTrace)
reTrace[i].append(hmmIO.int2str(i))
return reValue, reTrace
names = []
sentences = []
count = -1
# for sentence in range(1, 4):
timeBegin = time.time()
for i in range(1, len(length)):
# print "This sentence starts at ", keyOrder[count+1]
# print "Sentence Length = ", length[i] - length[i - 1]
# print "Toatal Length = ", length[i]
timeStart = time.time()
trace = []
value = []
for index in range(length[i] - length[i - 1]):
count += 1
if index == 0:
for j in range(0, 48):
value.append(
math.log(observationProb[keyOrder[count]][j]))
trace.append([hmmIO.int2str(j)])
else:
value, trace = forward(keyOrder[count], value, trace)
timeEnd = time.time()
# print "Time Cost = ", timeEnd - timeStart
# print "Total Time = ", timeEnd - timeBegin
maxIndex = np.argmax(value)
# print "max trace: ", trace[maxIndex]
# print 'max value = ', value[maxIndex], '\n', '\n'
s = re.split('_', keyOrder[count])[:2]
names.append(s[0] + '_' + s[1])
sentences.append(trace[maxIndex])
#preValue = [1.3212,....]
#preTrace = [sil, d, s, ...]
return names, sentences
def viterbi():
observationProb, keyOrder, length = hmmIO.readFile('../data/rnnTest.prb')
outputData = {}
trainLabel = hmmIO.readTrainLabel('../data/label/train.lab')
transMatrix = hmmFunc.transProb(trainLabel)
def forward(key, preValue, preTrace):
reValue = []
reTrace = []
for i in range(0,48):
tempValue = [preValue[prei] +
math.log(transMatrix[prei][i]) +
math.log(observationProb[key][i]) for prei in range(0,48)]
maxIndex = np.argmax(tempValue)
reValue.append(tempValue[maxIndex])
tempTrace = preTrace[maxIndex][:]
reTrace.append(tempTrace)
reTrace[i].append(hmmIO.int2str(i))
return reValue, reTrace
names = []
sentences = []
count = -1
# for sentence in range(1, 4):
timeBegin = time.time()
for i in range(1, len(length)):
# print "This sentence starts at ", keyOrder[count+1]
# print "Sentence Length = ", length[i] - length[i - 1]
# print "Toatal Length = ", length[i]
timeStart = time.time()
trace = []
value = []
for index in range(length[i] - length[i - 1]):
count += 1
if index == 0:
for j in range(0,48):
value.append(math.log(observationProb[keyOrder[count]][j]))
trace.append([hmmIO.int2str(j)])
else:
value, trace = forward(keyOrder[count], value, trace)
timeEnd = time.time()
# print "Time Cost = ", timeEnd - timeStart
# print "Total Time = ", timeEnd - timeBegin
maxIndex = np.argmax(value)
# print "max trace: ", trace[maxIndex]
# print 'max value = ', value[maxIndex], '\n', '\n'
s = re.split('_', keyOrder[count])[:2]
names.append(s[0] + '_' + s[1])
sentences.append(trace[maxIndex])
#preValue = [1.3212,....]
#preTrace = [sil, d, s, ...]
return names, sentences
def forward(key, preValue, preTrace):
reValue = []
reTrace = []
for i in range(0,48):
tempValue = [preValue[prei] +
math.log(transMatrix[prei][i]) +
math.log(observationProb[key][i]) for prei in range(0,48)]
maxIndex = np.argmax(tempValue)
reValue.append(tempValue[maxIndex])
tempTrace = preTrace[maxIndex][:]
reTrace.append(tempTrace)
reTrace[i].append(hmmIO.int2str(i))
return reValue, reTrace
def forward(key, preValue, preTrace):
reValue = []
reTrace = []
for i in range(0, 48):
tempValue = [
preValue[prei] + math.log(transMatrix[prei][i]) +
math.log(observationProb[key][i]) for prei in range(0, 48)
]
maxIndex = np.argmax(tempValue)
reValue.append(tempValue[maxIndex])
tempTrace = preTrace[maxIndex][:]
reTrace.append(tempTrace)
reTrace[i].append(hmmIO.int2str(i))
return reValue, reTrace