在cuda convnet 中加入一下脚本，可以在每个循环的训练中打断样本顺序

vola9527

于 2014-06-04 20:50:47 发布

阅读量537

点赞数

分类专栏：机器学习机器学习文章标签：神经网络

本文链接：https://blog.csdn.net/vola9527/article/details/28449615

版权

机器学习同时被 2 个专栏收录

27 篇文章 0 订阅

订阅专栏

机器学习

26 篇文章 1 订阅

订阅专栏

import os
import random
def recomposition(data_dir,batch_range):
    path=data_dir
    os.chdir(path)

    oldNames=[]
    newNames=[]
    tempName=random.sample(range(1000000,1500000),len(batch_range))
    
    for item in batch_range:
        temp='data_batch_'+str(item)
        oldNames.append(temp)
    newNames=oldNames[:]  
    random.shuffle(newNames)
    random.shuffle(newNames)
    print oldNames

    print newNames

    for item in range(len(oldNames)):
        tempPath=path+oldNames[item]
        if os.path.exists(tempPath):
            os.rename(oldNames[item],str(tempName[item]))
        else :
            print 'there is no path %s '% tempPath
    
    for item in range(len(tempName)):
        tempPath=path+str(tempName[item])
        if os.path.exists(tempPath):
            os.rename(str(tempName[item]),newNames[item])
        else :
            print 'there is no path %s '% tempPath