YM_labels_model.py 文件源码

def main():
    labels_uni = np.zeros([4716,1])
    with open(flags.FLAGS.src_path_1, "rt", encoding='utf-8') as csvfile:
        spamreader = csv.reader(csvfile)
        line_num = 0
        for row in spamreader:
            line_num += 1
            print('the '+str(line_num)+'th file is processing')
            if line_num==1:
                continue
            lbs = row[1].split()
            for i in range(0,len(lbs),2):
                labels_uni[int(lbs[i])] += 1
    np.savetxt('labels_model.out', labels_uni, delimiter=',')