MycroftAI
/
mycroft-precise
mirror of https://github.com/MycroftAI/mycroft-precise.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix database groups file when in subdirectory and formatting

pull/10/head
Matthew D. Scholefield 2018-04-12 18:21:24 -05:00
parent 69e8af5bb5
commit fbdca29949
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
precise/train_data.py
View File

@ -65,7 +65,7 @@ class TrainData:
raise RuntimeError('Database file does not exist: ' + db_file)
train_groups = {}
train_group_file = join(dirname(db_folder), db_file.replace('.txt', '') + '.groups.json')
train_group_file = join(db_file.replace('.txt', '') + '.groups.json')
if isfile(train_group_file):
with open(train_group_file) as f:
train_groups = json.load(f)

6
precise/vectorization.py
View File

@ -43,8 +43,10 @@ def vectorize(audio: np.ndarray) -> np.ndarray:
audio = audio[-pr.max_samples:]
features = vectorize_raw(audio)
if len(features) < pr.n_features:
features = np.concatenate(
[np.zeros((pr.n_features - len(features), len(features[0]))), features])
features = np.concatenate([
np.zeros((pr.n_features - len(features), len(features[0]))),
features
])
if len(features) > pr.n_features:
features = features[-pr.n_features:]