weighted hash err

DessimozLab · Aug 30, 2024 · 7b713bd · 7b713bd
1 parent e5b3ee7
commit 7b713bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/HogProf/lshbuilder.py b/src/HogProf/lshbuilder.py
@@ -147,7 +147,8 @@ def __init__(self,h5_oma=None,fileglob = None, taxa=None,masterTree=None, saving
         else:
             #load machine learning weights
             self.treeweights = treeweights
-        wmg = WeightedMinHashGenerator(3*len(self.taxaIndex), sample_size = numperm , seed=1)
+        tax_max = max(self.taxaIndex.values())+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