Method next_batch

tf_classify_demo/sample_data.py:176–200

获取一批样本数据

(self, count)

    def next_batch(self, count):
        """
        获取一批样本数据
        """
        x_s = []
        y_s = []
        if self.pos >= len(self.data):
            print("error")
            sys.exit(1)
        while count > 0:
            item = self.data[self.pos]
            vectors = item['vectors']
            local_group_id = item['local_group_id']
            x_array = np.zeros([self.dim_info.x_dim], dtype=np.float)
            y_array = np.zeros(self.maps.group_id_size(), dtype=np.float)
            y_array[local_group_id] = 1
            for word_id in vectors:
                vector = vectors[word_id]
                for index, weight in enumerate(vector):
                    x_array[word_id*self.dim_info.vec_dim+index] = weight
            x_s.append(x_array)
            y_s.append(y_array)
            self.pos = (self.pos + 1) % len(self.data)
            count = count - 1
        return x_s, y_s

    def test_sets(self):
        """