From 188133261f93f080cc04f6c5386595881ec34f63 Mon Sep 17 00:00:00 2001
From: Wang Bin <wangbin.zibo@gmail.com>
Date: Sat, 4 Apr 2015 15:26:26 +0800
Subject: [PATCH] small tweaks, added bench.sh for benchmark

---
 posseg/bench.sh      | 5 +++++
 posseg/prob_trans.go | 4 +++-
 2 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100755 posseg/bench.sh

diff --git a/posseg/bench.sh b/posseg/bench.sh
new file mode 100755
index 0000000..03c3a4c
--- /dev/null
+++ b/posseg/bench.sh
@@ -0,0 +1,5 @@
+#!/bin/sh
+
+go test -run=XXX -v -bench=BenchmarkViterbi -benchtime 10s  -benchmem -memprofile viterbi-mem.out -cpuprofile viterbi-cpu.out
+go tool pprof -png -output ~/tmp/viterbi-cpu.png posseg.test viterbi-cpu.out
+go tool pprof -png -output ~/tmp/viterbi-mem.png posseg.test viterbi-mem.out
diff --git a/posseg/prob_trans.go b/posseg/prob_trans.go
index 370bba7..08fcf74 100644
--- a/posseg/prob_trans.go
+++ b/posseg/prob_trans.go
@@ -4,13 +4,15 @@ import (
 	"math"
 )
 
+var inf = math.Inf(-1)
+
 type probTransMap map[string]float64
 
 func (m probTransMap) Get(key string) float64 {
 	if value, ok := m[key]; ok {
 		return value
 	} else {
-		return math.Inf(-1)
+		return inf
 	}
 }