-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmetrics_from_csv.py
More file actions
32 lines (30 loc) · 1.25 KB
/
metrics_from_csv.py
File metadata and controls
32 lines (30 loc) · 1.25 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import chainLLM
from langchain.embeddings import HuggingFaceEmbeddings
from sklearn.metrics.pairwise import cosine_similarity
import numpy as np
import pandas as pd
def similarity_score(sentence1,sentence2):
embeddings_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
embedding1 = embeddings_model.embed_query(sentence1)
embedding2 = embeddings_model.embed_query(sentence2)
embedding1 = np.array(embedding1).reshape(1, -1)
embedding2 = np.array(embedding2).reshape(1, -1)
similarity_score = cosine_similarity(embedding1, embedding2)[0][0]
similarity_score = round(similarity_score, 4)
return similarity_score
df = pd.read_csv('llm_Response.csv')
new_df = {
'Original Response': [],
'LLM Response': [],
'Similarity Score': []
}
for i in range(len(df)):
original_response = df['Original Response'][i]
llm_response = df['LLM Response'][i]
similarity = similarity_score(original_response,llm_response)
new_df['Original Response'].append(original_response)
new_df['LLM Response'].append(llm_response)
new_df['Similarity Score'].append(similarity)
print(f"Query {i} done.")
new_df = pd.DataFrame(new_df)
new_df.to_csv('evaluation_metrics_llm_org_similarity.csv', index=False)