This commit is contained in:
bwt09
2022-06-05 20:08:05 -07:00
parent 694084a2a2
commit 24cbeedfb1
2 changed files with 2 additions and 2 deletions

View File

@@ -75,7 +75,7 @@ class KnowledgeHarvester:
coded_ent_tuple = []
for b, ent in zip(bin_code, ent_tuple):
coded_ent_tuple.append(ent.title() if b == 1 else ent)
coded_ent_tuple.append(ent.title() if b == '1' else ent)
score = self.score_ent_tuple(ent_tuple=coded_ent_tuple)
if score > best_score:

View File

@@ -173,7 +173,7 @@
"There's no place like the <ENT0> for <ENT1> because you have access to all the equipment you need to get a good workout ."
]
},
"synonym": {
"antonym": {
"init_prompts": [
"<ENT0> is the opposite of <ENT1> ."
],