Tensorflow读取pandas数据训练DNN模型(心脏病预测(仅做深度学习训练参考,不可当成医学参考))

本案例使用Python的tensorflow、jupyternotebook、pandas进行训练学习
1.使用Pandas读取数据

import pandas as pd
import tensorflow as tf
df=pd.read_csv("./datas/heart/heart.csv")
df.head()

在这里插入图片描述

df.dtypes

在这里插入图片描述

#把thal列变成数字编码
df['thal']=pd.Categorical(df['thal'])
df['thal']=df['thal'].cat.codes
df.head()

在这里插入图片描述
2.将数据加载到tf.data.Dataset

#要预测的目标,这是一个二分类问题
target=df.pop('target')
df.head()

在这里插入图片描述

target.head()

在这里插入图片描述

df.head().values

在这里插入图片描述

df.shape
target.head().values
#构建dataset,其实是把pandas数据转换成numpy数组进行转换的
dataset=tf.data.Dataset.from_tensor_slices((df.values,target.values))
for features,label in dataset.take(5):
    print('Features:{},Label:{}'.format(features,label))

在这里插入图片描述

#Shuffle and batch the dataset
train_dataset=dataset.shuffle(len(df)).batch(1)

3.创建并训练模型

model=tf.keras.Sequential([
    tf.keras.layers.Dense(10,input_shape=(df.shape[1],)),
    tf.keras.layers.Dense(10,activation='relu'),
    tf.keras.layers.Dense(1)
])
model.compile(optimizer='adam',
             loss=tf.keras.losses.BinaryCrossentropy(from_logits=True),
             metrics=['accuracy'])
 model.summary()            

在这里插入图片描述

model.fit(train_dataset,epochs=15)

在这里插入图片描述

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值