Tjiho/ratatouille
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add fuzzy and text-classification light model #1

Merged
Tjiho merged 18 commits from fuzzy into master 2025-02-01 01:35:11 +01:00
Conversation 0 Commits 18 Files changed 11 +4 -3
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit f3afb19ae5 - Show all commits

Fix fuzzy for musique

Prev Next
tjiho 2024-11-10 20:13:21 +01:00
commit f3afb19ae5

7
src/fuzzy.py
View file

@ -35,7 +35,8 @@ SENTENCES_MUSIQUE_GENRE_SHORT = compute_sentences_musique_genre_short()
def fuzz_predict(text): def fuzz_predict(text):
choices = SENTENCES_HOUR + SENTENCES_LAMP_ON + SENTENCES_LAMP_OFF choices = SENTENCES_HOUR + SENTENCES_LAMP_ON + SENTENCES_LAMP_OFF + \
SENTENCES_MUSIQUE_GENRE_LONG + SENTENCES_MUSIQUE_GENRE_SHORT
result = process.extractOne( result = process.extractOne(
text, choices, scorer=fuzz.WRatio, processor=utils.default_process) text, choices, scorer=fuzz.WRatio, processor=utils.default_process)
@ -50,10 +51,10 @@ def fuzz_predict(text):
return {'intentName': 'LightOff', 'intentArg': [find_matching(ROOMS, text)]} return {'intentName': 'LightOff', 'intentArg': [find_matching(ROOMS, text)]}
if choosen_sentence in SENTENCES_MUSIQUE_GENRE_LONG: if choosen_sentence in SENTENCES_MUSIQUE_GENRE_LONG:
return {'intentName': 'LightOff', 'intentArg': [find_matching(MUSIQUE_GENRE_SHORT, text)]} return {'intentName': 'PlayMusicGenre', 'intentArg': [find_matching(MUSIQUE_GENRE_SHORT, text)]}
if choosen_sentence in SENTENCES_MUSIQUE_GENRE_SHORT: if choosen_sentence in SENTENCES_MUSIQUE_GENRE_SHORT:
return {'intentName': 'LightOff', 'intentArg': [find_matching(MUSIQUE_GENRE_SHORT, text)]} return {'intentName': 'PlayMusicGenre', 'intentArg': [find_matching(MUSIQUE_GENRE_SHORT, text)]}
return {'intentName': 'Unknown'} return {'intentName': 'Unknown'}