Search code examples
pythontensorflowio

Python_io in tensorflow


I'm having trouble working with tensorflow. I want to use TFRecordWriter() as below:

with tf.python_io.TFRecordWriter(testing_filename) as tfrecord_writer:
    # do sth

but I get the error:

AttributeError: module 'tensorflow' has no attribute 'python_io'

I'm working with tensorflow 1.2 and python 3.

How can I fix the problem?

Thanks.


Solution

  • The problem in the (python_io) :

    with tf.python_io.TFRecordWriter(testing_filename) as tfrecord_writer:
    # do sth
    

    changed it to:

    with tf.io.TFRecordWriter(testing_filename) as tfrecord_writer:
    # do sth