summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/tools/ai/core.ts8
1 files changed, 7 insertions, 1 deletions
diff --git a/src/tools/ai/core.ts b/src/tools/ai/core.ts
index 25bb807bba..5dcce26264 100644
--- a/src/tools/ai/core.ts
+++ b/src/tools/ai/core.ts
@@ -20,7 +20,13 @@ export default class Categorizer {
}
private tokenizer(text: string) {
- return this.mecab.wakachiSync(text);
+ const tokens = this.mecab.parseSync(text)
+ // 名詞だけに制限
+ .filter(token => token[1] === '名詞')
+ // 取り出し
+ .map(token => token[0]);
+
+ return tokens;
}
public async init() {