GCA-LSTM代码笔记

 

1.tf.data.TFRecordDataset

        #读取tf数据
	tfrecord_dataset = tf.data.TFRecordDataset(dataset)

	#取buffer_size个数据放入缓冲区，打乱顺序
	tfrecord_dataset = tfrecord_dataset.shuffle(buffer_size=1000)

	##在_parse_函数中对每一条数据进行解析、处理
	tfrecord_dataset = tfrecord_dataset.map(lambda x: _parse_(x)).shuffle(True)

	#从buffer中取4条数据
	data = tfrecord_dataset .batch(4)

	# repeat()为空表示一直重复下去
	if mode == 'test':
	    tfrecord_dataset = tfrecord_dataset.repeat()

	# repeat(1)表示只读取全部数据一次
	else:
	    tfrecord_dataset = tfrecord_dataset.repeat(1)

	#创建一个迭代器
	tfrecord_dataset = tfrecord_dataset.make_one_shot_iterator()

	#读取下一条数据

        next_element = tfrecord_dataset.get_next()