summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--lib/myconfig.py7
-rw-r--r--prepare.py2
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)