[WEI] For each bus, choose a random word which score is higher than the mid score

Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>
This commit is contained in:
Yohann D'ANELLO 2021-09-05 18:39:03 +02:00
parent dbc7b3444b
commit d888d5863a
Signed by: ynerant
GPG Key ID: 3A75C55819C8CF85
1 changed files with 7 additions and 3 deletions

View File

@ -49,9 +49,13 @@ class WEISurveyForm2021(forms.Form):
# Update seed
rng.randint(0, information.step)
preferred_words = {bus: [word for word in WORDS if bus.size > 0
and WEIBusInformation2021(bus).scores[word] >= 50]
for bus in WEISurveyAlgorithm2021.get_buses()}
buses = WEISurveyAlgorithm2021.get_buses()
scores = sum((list(WEIBusInformation2021(bus).scores.values()) for bus in buses), [])
average_score = sum(scores) / len(scores)
preferred_words = {bus: [word for word in WORDS
if WEIBusInformation2021(bus).scores[word] >= average_score]
for bus in buses}
while words is None or len(set(words)) != len(words):
# Ensure that there is no the same word 2 times
words = [rng.choice(words) for _ignored2, words in preferred_words.items()]