re-factor code

2 files changed, 8 insertions, 1 deletions

diff --git a/lib/myconfig.py b/lib/myconfig.py
index 5db51e2..432e89b 100644
--- a/lib/myconfig.py
+++ b/lib/myconfig.py
@@ -10,6 +10,7 @@ class MyConfig:
                            'PruneEpoch': 1, \
                            'EvaluateEpoch': 1, \
                            'PrepareEpoch': 2, \
+                           'PopulateEpoch': 3, \
                            }
 
     def getEpochs(self):
@@ -114,6 +115,12 @@ class MyConfig:
     def getWordRecognizerDir(self):
         return self.m_trainer_dir + os.sep + 'words'
 
+    def getNgramFileName(self, length):
+        return str(length) + '-gram.db'
+
+    def getWordSep(self):
+        return " "
+
     def getMaximumCombineNumber(self):
         N = 5
         assert N >= 2, 'at least bi-gram'
diff --git a/prepare.py b/prepare.py
index cd6e4e6..7d45bea 100644
--- a/prepare.py
+++ b/prepare.py
@@ -43,7 +43,7 @@ def createSqliteDatabases(onedir):
 
     #create databases
     for i in range(1, N + 1):
-        filename = str(i) + '-gram.db';
+        filename = config.getNgramFileName(i)
         filepath = onedir + os.sep + filename
         print(filepath)