def closest_contexts(self, w, n=10): scores = self.ec.m.dot(self.ew.represent(w)) pairs = zip(scores, self.ec.iw)[1:] return heapq.nlargest(n, pairs)