数据洗牌

在实验过程中,可能出现海量数据但是过拟合的现象,这种情况可能是由于train数据和val/test的数据差别较大引起的。在这种情况下可以进行数据洗牌,即通过重新调整数据,将数据重新打乱按照之前的分配比例重新分割数据,形成新的train/val/test这种方式不会损坏数据原有的结构,可能消除过拟合的现象。

详细代码如下:

#coding=utf-8
import numpy as np
import os
import glob
import time
from random import shuffle
import os.path
import shutil

path = ''
path_new = ''


#创建新的文件夹存放
def new_dir(_new_dir):
    """new paht save image"""
    if not os.path.exists(_new_dir):
        os.makedirs(_new_dir)
        #print _new_dir+" has been created"


def moveFileto(oldDir, newDir, falg):
    #复制文件并重命名
    shutil.copy(oldDir, newDir+'/'+falg)


def save_npy(path_load):

    data_files = glob.glob(path_load + '*/' + '/*.npy')
    list = []
    print (path_load)
    for j, i in enumerate(data_files):
        list.append(i)
    shuffle(list)
    for j, i in enumerate(list):
        paths = i.split('/')[7:]
        #print (paths)
        if j < 50:
            new_path = path_new + '/' + paths[0] + '/val'
            falg = paths[0]+'_'+str(j).zfill(5)+'.npy'
        elif j >= 50 and  j < 100:
            new_path = path_new + '/' + paths[0] + '/test'
            falg = paths[0] + '_' + str(j - 49).zfill(5) + '.npy'
        else:
            new_path = path_new + '/' + paths[0] + '/train'
            falg = paths[0]+'_'+str(j-99).zfill(5)+'.npy'
        new_dir(new_path)
        moveFileto(i, new_path, falg)


def load_path():

    with open('./label_sorted.txt') as myfile:
        data_dir = myfile.read().splitlines()

    for j in range(len(data_dir)):
        #print data_dir[j]
        path_load = path + '/'+data_dir[j] +'/'
        #print path_load
        save_npy(path_load)


load_path()

 

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值