def proba(self, features, params): """ return a categorical probability distribution over the vocabulary """ product = np.dot(features, params) return softmax(product, self.T)