diff --git a/decision_layer/decision_layer.py b/decision_layer/decision_layer.py index af2808d1f7a0f35270b9addf3253e575d3e3b5ea..25130dfde60c53cb2764cde68228106e9b182f60 100644 --- a/decision_layer/decision_layer.py +++ b/decision_layer/decision_layer.py @@ -55,9 +55,7 @@ class DecisionLayer: # create query vector xq = np.array(self.encoder([text])) xq = np.squeeze(xq) # Reduce to 1d array. - sim = np.dot(self.index, xq.T) / (norm(self.index, axis=1)*norm(xq.T)) - # get indices of top_k records top_k = min(top_k, sim.shape[0]) idx = np.argpartition(sim, -top_k)[-top_k:]