将Pandas数据框转换为Tensorflow数据集

时间:2019-07-19 19:02:38

标签: python python-3.x pandas tensorflow tensor

我正在尝试将Pandas数据框转换为Tensorflow数据集。 DataFrame包括三列:

  1. 目标-二进制[1/0]
  2. 功能1-整数列表。
  3. 功能2-整数列表。

Feature1和Feature2是基于Tensorflow的TokenTextEncoder的索引列表。下面的第一行示例:

目标Feature1 Feature2   1 [1,2,3,5] [6,7,9,10]

目标是基于两个功能集创建一个二进制分类。

我曾尝试将列表转换为numpy数组,但仍然遇到相同的错误。

target = df.pop('Target')
dataset = tf.data.Dataset.from_tensor_slices((df.values, target.values))

TypeError:预期的二进制或Unicode字符串,得到[399,309,34,383,281]

0 个答案:

没有答案