constructor.py 文件源码

def next_batch(self, batch_size=10, start=-1, end=-1, nth=-1):
        data = np.empty((0, self._data_len), float)
        label = np.empty((0, self._class_len), int)

        with open(PACK_PATH+"/dataset/"+str(self._who_am_i)+".csv") as f:
            lines = f.readlines()

        if(nth == -1):
            if((start == -1) and (end == -1)):
                datas = random.sample(lines, batch_size)
            else:
                datas = lines[start:end]
        else:
            datas = []
            datas.append(lines[nth])

        for d in datas:
            sv_data = d.split(',')
            tmp_label = sv_data[0]
            tmp_data = sv_data[1:len(sv_data)-1]

            tmp_data = np.asarray(tmp_data).reshape((1, len(tmp_data)))

            label = np.append(label, np.eye(self._class_len)[int(np.asfarray(tmp_label))].reshape(1, self._class_len), axis=0)
            data = np.append(data, tmp_data, axis=0)

        return data, label