File tree 2 files changed +4
-4
lines changed
2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -182,7 +182,7 @@ def get_docs_embedding(
182
182
def cosine_similarity (self , a : List [float ], b : List [float ]) -> float :
183
183
return round (np .dot (a , b ) / (np .linalg .norm (a ) * np .linalg .norm (b )), 4 )
184
184
185
- def similart_tables_based_on_few_shot_examples (self , df : pd .DataFrame ) -> List [str ]:
185
+ def similar_tables_based_on_few_shot_examples (self , df : pd .DataFrame ) -> List [str ]:
186
186
most_similar_tables = set ()
187
187
if self .few_shot_examples is not None :
188
188
for example in self .few_shot_examples :
@@ -224,7 +224,7 @@ def _run(
224
224
)
225
225
df = df .sort_values (by = "similarities" , ascending = True )
226
226
df = df .tail (TOP_TABLES )
227
- most_similar_tables = self .similart_tables_based_on_few_shot_examples (df )
227
+ most_similar_tables = self .similar_tables_based_on_few_shot_examples (df )
228
228
table_relevance = ""
229
229
for _ , row in df .iterrows ():
230
230
table_relevance += f'Table: `{ row ["table_name" ]} `, relevance score: { row ["similarities" ]} \n '
Original file line number Diff line number Diff line change @@ -246,7 +246,7 @@ def get_docs_embedding(
246
246
def cosine_similarity (self , a : List [float ], b : List [float ]) -> float :
247
247
return round (np .dot (a , b ) / (np .linalg .norm (a ) * np .linalg .norm (b )), 4 )
248
248
249
- def similart_tables_based_on_few_shot_examples (self , df : pd .DataFrame ) -> List [str ]:
249
+ def similar_tables_based_on_few_shot_examples (self , df : pd .DataFrame ) -> List [str ]:
250
250
most_similar_tables = set ()
251
251
if self .few_shot_examples is not None :
252
252
for example in self .few_shot_examples :
@@ -288,7 +288,7 @@ def _run(
288
288
)
289
289
df = df .sort_values (by = "similarities" , ascending = True )
290
290
df = df .tail (TOP_TABLES )
291
- most_similar_tables = self .similart_tables_based_on_few_shot_examples (df )
291
+ most_similar_tables = self .similar_tables_based_on_few_shot_examples (df )
292
292
table_relevance = ""
293
293
for _ , row in df .iterrows ():
294
294
table_relevance += f'Table: `{ row ["table_name" ]} `, relevance score: { row ["similarities" ]} \n '
You can’t perform that action at this time.
0 commit comments