मैं जावा में निम्नलिखित कोड कैसे लिख सकता हूं? अगर मेरे पास जावा में रिकॉर्ड्स/डिक्ट्स की सूची है तो मैं उन्हें tfrecords में लिखने के लिए बीम कोड कैसे लिख सकता हूं जहां tf.train.Examples क्रमबद्ध हैं। अजगर के साथ ऐसा करने के लिए बहुत सारे उदाहरण हैं, नीचे अजगर में एक उदाहरण है, मैं जावा में एक ही तर्क कैसे लिख सकता हूं?

import tensorflow as tf
import apache_beam as beam
from apache_beam.runners.interactive import interactive_runner
from apache_beam.coders import ProtoCoder

class Foo(beam.DoFn):
  def process(self, element, *args, **kwargs):
    import tensorflow as tf

    foo = element.get('foo')
    bar = element.get('bar')

    feature = {
      "foo":
        tf.train.Feature(bytes_list=tf.train.BytesList(value=[foo.encode('utf-8')])),
      "bar":
        tf.train.Feature(bytes_list=tf.train.BytesList(value=[bar.encode('utf-8')]))
    }
    example_proto = tf.train.Example(features=tf.train.Features(feature=feature))
    yield example_proto

p = beam.Pipeline(runner=interactive_runner.InteractiveRunner())

records = p | "Create records" >> beam.Create([{'foo': 'abc', 'bar': 'pqr'} for _ in range(10)])
tf_examples = records | "Convert to tf examples" >> beam.ParDo(Foo())
tf_examples | "Dump Records" >> beam.io.WriteToTFRecord(file_path_prefix="./output/data-",
                                                    coder=ProtoCoder(tf.train.Example()),
                                                    file_name_suffix='.tfrecord', num_shards=2)

p.run()
0
bruce_wayne 16 अप्रैल 2020, 13:08

1 उत्तर

सबसे बढ़िया उत्तर

मैंने जावा के साथ इसका प्रयास किया है लेकिन मुझे अभी भी कुछ समस्याएं मिल रही हैं, नए प्रश्न का लिंक यहां है बीम पाइपलाइन से tfrecords लिखें?

0
Jayendra Parmar 17 अप्रैल 2020, 11:13