Jan Mühlnikel commited on
Commit
6b7c543
·
1 Parent(s): 8250706

experiment

Browse files
Files changed (1) hide show
  1. functions/calc_matches.py +8 -6
functions/calc_matches.py CHANGED
@@ -31,21 +31,23 @@ def calc_matches(filtered_df, project_df, similarity_matrix, top_x):
31
 
32
  # Prepare the result with row and column indices from original dataframes
33
  top_15_matches = []
 
 
34
  for value, row, col in zip(top_15_values, top_15_2d_indices[0], top_15_2d_indices[1]):
35
  original_row_index = project_indices[row]
36
  original_col_index = filtered_indices[col]
 
 
37
  top_15_matches.append((value, original_row_index, original_col_index))
38
 
39
  st.write(top_15_matches)
40
 
41
 
42
- """
43
- p1_df = filtered_df.loc[top_col_indices].copy()
44
- p1_df['similarity'] = top_values
45
 
46
- p2_df = project_df.loc[top_row_indices].copy()
47
- p2_df['similarity'] = top_values
48
  print("finished calc matches")
49
 
50
  return p1_df, p2_df
51
- """
 
31
 
32
  # Prepare the result with row and column indices from original dataframes
33
  top_15_matches = []
34
+ org_rows = []
35
+ org_cols = []
36
  for value, row, col in zip(top_15_values, top_15_2d_indices[0], top_15_2d_indices[1]):
37
  original_row_index = project_indices[row]
38
  original_col_index = filtered_indices[col]
39
+ org_rows.append(original_row_index)
40
+ org_cols.append(original_col_index)
41
  top_15_matches.append((value, original_row_index, original_col_index))
42
 
43
  st.write(top_15_matches)
44
 
45
 
46
+ p1_df = filtered_df.loc[org_cols].copy()
47
+ p1_df['similarity'] = top_15_values
 
48
 
49
+ p2_df = project_df.loc[org_rows].copy()
50
+ p2_df['similarity'] = top_15_values
51
  print("finished calc matches")
52
 
53
  return p1_df, p2_df