ave.py 文件源码

def print_results(file, out_file, file_type):
    """
    file:       word embedding file
    out_file:   if provided, where to write results
    file_type:  word2vec, glove, tang, or bin
    """

    names, results, dim = test_embeddings(file, file_type)

    table_data = [[name] + result for name, result in zip(names, results)]
    table = tabulate.tabulate(table_data,
                              headers=['dataset', 'acc', 'prec', 'rec', 'f1'],
                              tablefmt='simple', floatfmt='.3f')

    if out_file:
        with open(out_file, 'a') as f:
            f.write('\n')
            f.write('+++Average word vectors+++\n')
            f.write(table)
            f.write('\n')
    else:
        print()
        print('+++Average word vectors+++')
        print(table)