Jan Mühlnikel commited on
Commit
ac6359f
1 Parent(s): 5729f8c

added float16 to matrix

Browse files
Files changed (1) hide show
  1. similarity_page.py +2 -2
similarity_page.py CHANGED
@@ -32,7 +32,7 @@ def get_process_memory():
32
  @st.cache_data
33
  def load_sim_matrix():
34
  loaded_matrix = load_npz("src/extended_similarities.npz")
35
- dense_matrix = loaded_matrix.toarray()
36
 
37
  return dense_matrix
38
 
@@ -40,7 +40,7 @@ def load_sim_matrix():
40
  @st.cache_data
41
  def load_nonsameorga_sim_matrix():
42
  loaded_matrix = load_npz("src/extended_similarities_nonsimorga.npz")
43
- dense_matrix = loaded_matrix.toarray()
44
 
45
  return dense_matrix
46
 
 
32
  @st.cache_data
33
  def load_sim_matrix():
34
  loaded_matrix = load_npz("src/extended_similarities.npz")
35
+ dense_matrix = loaded_matrix.toarray().astype('float16')
36
 
37
  return dense_matrix
38
 
 
40
  @st.cache_data
41
  def load_nonsameorga_sim_matrix():
42
  loaded_matrix = load_npz("src/extended_similarities_nonsimorga.npz")
43
+ dense_matrix = loaded_matrix.toarray().astype('float16')
44
 
45
  return dense_matrix
46