xfnw/min
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

use multiple words in sentance generation when cheapie

Browse source
This commit is contained in:
xfnw 2021-11-11 19:43:03 -05:00
parent b5afc6135a
commit ae7b469676
1 changed files with 20 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
modules/nlp.py
View file

@ -2,23 +2,33 @@ from bot import *
import dataset, random, time, re
def get(l,i):
try:
if i <= len(l) and i >= 0:
return l[i]
else:
return ''
except IndexError:
return ''
async def rec(self, m):
prew = shared.db['prew']
noch = shared.db['noun']
beg = shared.db['beg']
end = shared.db['end']
pre = ''
words = m.split(' ')
if words[0] == 'admin':
if words[0] == 'admin' or len(words) < 2:
return
for w in words:
if pre == '':
beg.insert(dict(word=w))
else:
prew.insert_ignore(dict(pre=pre, pro=w),['id'])
pre = w
beg.insert(dict(word=words[0]))
end.insert(dict(word=words[-1]))
for w in range(len(words)):
if w > 0:
prew.insert_ignore(dict(pre3=get(words,w-3), pre2=get(words,w-2), pre=get(words,w-1), pro=get(words,w), pro2=get(words,w+1), pro3=get(words,w+2)),['id'])
noch.insert(dict(word=w))
end.insert(dict(word=pre))
async def getNoun(self, words, c):
if c in shared.cstate: