如何将熊猫数据框转换为张量流数据集？

Question

我正在将 csv 文件读入 pandas 数据框。

train_data = pd.read_csv('mnist_test.csv');

样本数据

   label  pixel1  pixel2  pixel3  ...  pixel781  pixel782  pixel783  pixel784
0      6     149     149     150  ...       106       112       120       107
1      5     126     128     131  ...       184       184       182       180
2     10      85      88      92  ...       226       225       224       222
3      0     203     205     207  ...       230       240       253       255
4      3     188     191     193  ...        49        46        46        53

如何将此数据帧转换为张量流数据集。

Answer 1

import tensorflow as tf
ds = tf.data.Dataset.from_tensor_slices(dict(train_data))

有关详细信息，请参阅 tensorflow.org/tutorials/load_data/pandas_dataframe。

Answer 2

为了完整起见，

import tensorflow as tf
ds = tf.data.Dataset.from_tensor_slices(train_data.to_dict(orient="list"))
print(ds)
TensorSliceDataset element_spec={'label': TensorSpec(shape=(), dtype=tf.int32, name=None), ...}