YM_labels_model.py 文件源码

python
阅读 25 收藏 0 点赞 0 评论 0

项目:youtube-8m 作者: wangheda 项目源码 文件源码
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=',')
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号