擅长:python、mysql、java
<p>数据集的输入管道始终跟踪到图形中(就像使用了<a href="https://www.tensorflow.org/api_docs/python/tf/function" rel="nofollow noreferrer">^{<cd1>}</a>),以使其更快,这意味着,除其他外,您不能使用<code>.numpy()</code>。但是,您可以使用<a href="https://www.tensorflow.org/api_docs/python/tf/numpy_function" rel="nofollow noreferrer">^{<cd3>}</a>在图形中以NumPy数组的形式访问数据:</p>
<pre class="lang-py prettyprint-override"><code>def transform(example):
# example will now by a NumPy array
str_example = example.decode("utf-8")
json_example = json.loads(str_example)
overall = json_example.get('overall', None)
text = json_example.get('reviewText', None)
return (overall, text)
line_dataset = tf.data.TextLineDataset(filenames = [file_path])
line_dataset = line_dataset.map(
lambda row: tf.numpy_function(transform, row, (tf.float32, tf.string)))
for example in line_dataset.take(5):
print(example)
</code></pre>