TFRecorder用于将PandasDataFrame或CSV文件转换为TFRecord。TFRecord是一种简单的二进制文件格式。
目前，TFRecorder支持以“图像csv格式”存储的数据，类似于GCPAutoMLVision。
用法示例FromPandasDataFrame本地运行
importpandasaspdimporttfrecorderdf=pd.read_csv(...)df.tensorflow.to_tfr(output_dir='gs://my/bucket')在CloudDataflow运行
importpandasaspdimporttfrecorderdf=pd.read_csv(...)df.tensorflow.to_tfr(output_dir='gs://my/bucket',runner='DataFlowRunner',project='my-project',region='us-central1')FromCSV使用Python解释器
importtfrecordertfrecorder.create_tfrecords(input_data='/path/to/data.csv',output_dir='gs://my/bucket')使用命令行
tfrecordercreate-tfrecords\--input_data=/path/to/data.csv\--output_dir=gs://my/bucket
评论