input vec

DessimozLab · Aug 30, 2024 · d550320 · d550320
1 parent 7b713bd
commit d550320
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/HogProf/lshbuilder.py b/src/HogProf/lshbuilder.py
@@ -148,7 +148,7 @@ def __init__(self,h5_oma=None,fileglob = None, taxa=None,masterTree=None, saving
             #load machine learning weights
             self.treeweights = treeweights
         tax_max = max(self.taxaIndex.values())+1
-        wmg = WeightedMinHashGenerator(3*tax_max, sample_size = numperm , seed=1)
+        wmg = WeightedMinHashGenerator(3*tax_max , sample_size = numperm , seed=1)
         with open( self.saving_path  + 'wmg.pkl', 'wb') as wmgout:
             wmgout.write( pickle.dumps(wmg))
         self.wmg = wmg

diff --git a/src/HogProf/utils/hashutils.py b/src/HogProf/utils/hashutils.py
@@ -85,7 +85,7 @@ def hash_tree(tp , taxaIndex , treeweights , wmg , lossonly = False , duplonly =
     input_vec = list(hog_matrix_weighted.flatten())
 
     if wmg.dim == len(input_vec):
-        weighted_hash = wmg.minhash()
+        weighted_hash = wmg.minhash(input_vec)
         return  hog_matrix_binary , weighted_hash
 
     else: