작성 방법
1.
Feature(Column)를 Dict 형태로 한 줄씩 묶는다.
2.
Dict를 tf.train.Example Object로 만든다.
3.
tf.io.TFRecordWriter 를 이용해 파일로 작성한다.
# Write the records to a file.
with tf.io.TFRecordWriter(example_path) as file_writer:
for _ in range(4):
x, y = np.random.random(), np.random.random()
record_bytes = tf.train.Example(features=tf.train.Features(feature={
"x": tf.train.Feature(float_list=tf.train.FloatList(value=[x])),
"y": tf.train.Feature(float_list=tf.train.FloatList(value=[y])),
})).SerializeToString()
file_writer.write(record_bytes)
Python
복사
디테일
tf.train.Example
•
한 Row씩 저장된 Feature Dict를 담는 객체이다.
•
사용 가능한 데이터 타입은 다음과 같으며, 데이터 타입에 따라 적절히 매핑해주어야 한다.
•
tf.train.BytesList (string, byte)
•
tf.train.FloatList (float, double)
•
tf.train.Int64List (bool, enum, int32, uint32, int64, uint64)