예제 #1
0
def gen_level():
    print("Initializing DAG...", file=sys.stderr)
    dag = Dag.loads_from(ID_HIERARCHY_FILE)
    print("Generating level.txt ...", file=sys.stderr)
    fout = open(LEVEL_FILE, 'w')
    level_list = dag.level_list()
    for i in range(len(level_list)):
        fout.write("{0}\t{1}".format(i, level_list[i]) + "\n")
    fout.close()
예제 #2
0
def gen_level():
    print("Initializing DAG...", file=sys.stderr)
    dag = Dag.loads_from(ID_HIERARCHY_FILE)
    print("Generating level.txt ...", file=sys.stderr)
    fout = open(LEVEL_FILE, 'w')
    level_list = dag.level_list()
    for i in range(len(level_list)):
        fout.write("{0}\t{1}".format(i, level_list[i]) + "\n")
    fout.close()
예제 #3
0
def category2ancestor():
    print("Initializing DAG...", file=sys.stderr)
    dag = Dag.loads_from(ID_HIERARCHY_FILE)
    print("Generating entity2ancestor.txt ...", file=sys.stderr)
    fin = open(ENTITY2CATEGORY_FILE)
    fout = open(ENTITY2ANCESTOR_FILE, 'w')
    count = 0
    for line in fin:
        if count % 1000 == 0:
            print("{0}\r".format(count), file=sys.stderr, end="")
        info = line.split('\t')
        entity = info[0]
        clist = [int(c) for c in info[1:]]
        stats = dag.ancestor_path(clist)
        fout.write("\t".join([entity] + ["{0}:{1}".format(c[0], c[1]) for c in stats]) + "\n");
        count += 1
    fout.close()
    fin.close()
예제 #4
0
def category2ancestor():
    print("Initializing DAG...", file=sys.stderr)
    dag = Dag.loads_from(ID_HIERARCHY_FILE)
    print("Generating entity2ancestor.txt ...", file=sys.stderr)
    fin = open(ENTITY2CATEGORY_FILE)
    fout = open(ENTITY2ANCESTOR_FILE, 'w')
    count = 0
    for line in fin:
        if count % 1000 == 0:
            print("{0}\r".format(count), file=sys.stderr, end="")
        info = line.split('\t')
        entity = info[0]
        clist = [int(c) for c in info[1:]]
        stats = dag.ancestor_path(clist)
        fout.write("\t".join([entity] +
                             ["{0}:{1}".format(c[0], c[1])
                              for c in stats]) + "\n")
        count += 1
    fout.close()
    fin.close()