87f1ec81862468bd6381fd5a5d1579861a2a27d4,test/basic_test.py,,pmi_test,#Any#,194

Before Change

        alpha=0.0001,
        verbose=True)

    for pair, pmi in sorted(pmi_dok.items(), key=lambda x:-x[1])[100:110]:
        pair_ = (idx2vocab[pair[0]], idx2vocab[pair[1]])
        print("pmi {} = {:.3f}".format(pair_, pmi))
    print("computed PMI")

After Change

    print("col  shape = {}".format(cols.shape))
    print("data shape = {}".format(data.shape))

    for indpt in data.argsort()[-150:-100]:
        i = rows[indpt]
        j = cols[indpt]
        pair = (idx2vocab[i], idx2vocab[j])
        value = data[indpt]
        print("pmi {} = {:.3f}".format(pair, value))
    print("computed pmi")


def main():

Instances