diff options
Diffstat (limited to 'lualib/redis_scripts/bayes_learn.lua')
| -rw-r--r-- | lualib/redis_scripts/bayes_learn.lua | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/lualib/redis_scripts/bayes_learn.lua b/lualib/redis_scripts/bayes_learn.lua index b0ed1dd4b..bf2775349 100644 --- a/lualib/redis_scripts/bayes_learn.lua +++ b/lualib/redis_scripts/bayes_learn.lua @@ -31,12 +31,14 @@ for i, token in ipairs(input_tokens) do local tok1 = text_tokens[i * 2 - 1] local tok2 = text_tokens[i * 2] - if tok2 then - redis.call('HSET', token, 'tokens', string.format('%s:%s', tok1, tok2)) - else - redis.call('HSET', token, 'tokens', tok1) - end + if tok1 then + if tok2 then + redis.call('HSET', token, 'tokens', string.format('%s:%s', tok1, tok2)) + else + redis.call('HSET', token, 'tokens', tok1) + end - redis.call('ZINCRBY', prefix .. '_z', token, is_unlearn and -1 or 1) + redis.call('ZINCRBY', prefix .. '_z', token, is_unlearn and -1 or 1) + end end end
\ No newline at end of file |