本案例使用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)

在这里插入图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐