def __init__(self, name, data_files_pattern, dataset_class=TextLineDataset,
min_queue_examples=0, shuffle_size=None, padded_shapes=None, padded_values=None):
"""
:param name: name of the dataset.
:param str data_files_pattern: pattern of the data files
:param Dataset dataset_class: class to create the dataset with the files. By default is
TextLineDataset
:param int min_queue_examples: minimum number of examples to queue, this value should be
proportional to the ram of the computer. By default is 0
:param int shuffle_size: size of the buffer for shuffling, this value should be
proportional to the ram of the computer
:param List[tf.Tensor] padded_shapes: shape for padding the batch
:param tf.Tensor padded_values: values for the padding
"""
self.name = name
self.data_files_pattern = data_files_pattern
self.dataset_class = dataset_class
self.min_queue_examples = min_queue_examples
self.shuffle_size = shuffle_size
self.padded_shapes = padded_shapes
self.padded_values = padded_values
self._size = None
tf_dataset.py 文件源码
python
阅读 23
收藏 0
点赞 0
评论 0
评论列表
文章目录