(value)
| 321 | return features, label |
| 322 | |
| 323 | def parse_parquet(value): |
| 324 | tf.logging.info('Parsing {}'.format(filename)) |
| 325 | labels = [] |
| 326 | for i in range(0, len(LABEL_COLUMNS)): |
| 327 | labels.append(value.pop(LABEL_COLUMNS[i])) |
| 328 | label = tf.stack(labels, axis=1) |
| 329 | features = value |
| 330 | return features, label |
| 331 | |
| 332 | '''Work Queue Feature''' |
| 333 | if args.workqueue and not args.tf: |