在做数据分析时,经常用dataframe来存储和操作数据,但是TensorFlow训练时需要tensor格式的数据,下面分享一个将dataframe数据转为tensor的方法。
场景:
给定三个属性值‘性别’、‘年龄’、‘身高’,预测‘体重’。
(1)创建数据集,并保存为dataframe格式
(2)上述数据集中的feature和label是放在一起的,我们使用tf.constant来分别创建feature和label的tensor
至此,feature和label的tensor创建完成,可以进一步编写网络模型进行训练了。