Example #1
0
    def __init__(self):
        """
        InputTableDataset: derived from PadBoxSlotDataset.

        Examples:
            .. code-block:: python

            import paddle.fluid as fluid
            dataset = fluid.DatasetFactory().create_dataset("InputTableDataset")
        """
        self.proto_desc = data_feed_pb2.DataFeedDesc()
        self.proto_desc.pipe_command = "cat"
        self.dataset = core.Dataset("InputTableDataset")
        self.thread_num = 1
        self.filelist = []
        self.boxps = core.BoxPS(self.dataset)
        self.proto_desc.name = "InputTableDataFeed"
        self.fleet_send_batch_size = None
        self.is_user_set_queue_num = False
        self.queue_num = None
        self.parse_ins_id = False
        self.parse_content = False
        self.parse_logkey = False
        self.merge_by_sid = True
        self.enable_pv_merge = False
        self.merge_by_lineid = False
        self.fleet_send_sleep_seconds = None
 def __init__(self):
     """ Init. """
     # define class name here
     # to decide whether we need create in memory instance
     self.proto_desc = data_feed_pb2.DataFeedDesc()
     self.proto_desc.pipe_command = "cat"
     self.dataset = core.Dataset("MultiSlotDataset")
     self.thread_num = 0
Example #3
0
 def __init__(self, proto_file):
     self.proto_desc = data_feed_pb2.DataFeedDesc()
     with open(proto_file, 'r') as f:
         text_format.Parse(f.read(), self.proto_desc)
     if self.proto_desc.name == "MultiSlotDataFeed":
         self.__name_to_index = {
             slot.name: i
             for i, slot in enumerate(self.proto_desc.multi_slot_desc.slots)
         }
Example #4
0
 def __init__(self):
     """
     Initialize BoxPSDataset
     This class should be created by DatasetFactory
     """
     # define class name here
     # to decide whether we need create in memory instance
     self.proto_desc = data_feed_pb2.DataFeedDesc()
     self.proto_desc.pipe_command = "cat"
     self.dataset = core.Dataset("PadBoxSlotDataset")
     self.thread_num = 1
     self.filelist = []
     self.boxps = core.BoxPS(self.dataset)
     self.proto_desc.name = "SlotPaddleBoxDataFeed"
     self.fleet_send_batch_size = None
     self.is_user_set_queue_num = False
     self.queue_num = None
     self.parse_ins_id = False
     self.parse_content = False
     self.parse_logkey = False
     self.merge_by_sid = True
     self.enable_pv_merge = False
     self.merge_by_lineid = False
     self.fleet_send_sleep_seconds = None