Jan Mühlnikel
commited on
Commit
·
d7f99ce
1
Parent(s):
c8d1d9b
experiment
Browse files
functions/calc_matches.py
CHANGED
@@ -68,11 +68,14 @@ def calc_matches(filtered_df, project_df, similarity_matrix, top_x):
|
|
68 |
top_values = match_matrix.data[linear_indices]
|
69 |
|
70 |
# Create resulting dataframes with top matches and their similarity scores
|
|
|
71 |
p1_df = filtered_df.loc[top_indices[0]].copy()
|
72 |
p1_df['similarity'] = top_values
|
|
|
|
|
73 |
p2_df = project_df.loc[top_indices[1]].copy()
|
74 |
p2_df['similarity'] = top_values
|
75 |
-
|
76 |
print("finished calc matches")
|
77 |
|
78 |
return p1_df, p2_df
|
|
|
68 |
top_values = match_matrix.data[linear_indices]
|
69 |
|
70 |
# Create resulting dataframes with top matches and their similarity scores
|
71 |
+
st.write(top_indices)
|
72 |
p1_df = filtered_df.loc[top_indices[0]].copy()
|
73 |
p1_df['similarity'] = top_values
|
74 |
+
|
75 |
+
st.dataframe(p1_df)
|
76 |
p2_df = project_df.loc[top_indices[1]].copy()
|
77 |
p2_df['similarity'] = top_values
|
78 |
+
st.dataframe(p2_df)
|
79 |
print("finished calc matches")
|
80 |
|
81 |
return p1_df, p2_df
|