tensorflow实现逻辑回归

本文通过使用TensorFlow框架和sklearn的乳腺癌数据集,实现了对乳腺癌预测模型的训练。在数据预处理之后,采用sigmoid_cross_entropy_with_logits损失函数进行模型训练,并通过多次迭代达到了较好的预测效果。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

数据用的sklearn自带的乳腺癌数据集,中间写了好几次loss函数都出问题了,最后用的tf.nn.sigmoid_cross_entropy_with_logits(_sentinel=None, labels=None, logits=None, name=None)这个方法实现的。

代码如下:

from sklearn.model_selection import train_test_split
import tensorflow as tf
from sklearn.datasets import load_breast_cancer
import numpy as np
from sklearn.preprocessing import MinMaxScaler

#数据预处理

breast_cancer=load_breast_cancer()
label=breast_cancer.target.reshape([-1,1])
feature=breast_cancer.data
minmxx=MinMaxScaler([0,1])
minmxx.fit(feature)
feature=minmxx.transform(feature)

#tensorflow

w=tf.Variable(tf.random_normal([30,1]))
b=tf.Variable(tf.zeros([1,1]))
x=tf.placeholder(tf.float32)
y=tf.placeholder(tf.float32)
ye=tf.matmul(x,w)+b
pre=tf.sigmoid(ye)
#loss=-tf.reduce_mean(tf.square(y-pre))
loss=tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(labels=y,logits=pre))
train=tf.train.AdamOptimizer(0.1).minimize(loss)
init=tf.global_variables_initializer()
sess=tf.Session()
sess.run(init)

for step in range(20001):
    sess.run(train,feed_dict={x:feature,y:label})
    if step%2000==0:
        print(sess.run([loss],feed_dict={x:feature,y:label}))

#计算测试精度

pred=sess.run(pre,feed_dict={x:feature,y:label})

from sklearn import metrics
print(metrics.classification_report(label,pred.round()))

最后计算结果还可以。刚开始学的时候总想着上手去写,所以过程中遇到了很多困难,比如矩阵乘法问题,loss总是不收敛、loss会出现nan的问题,也都找到了解决的办法,感觉是对于这个框架慢慢的逐渐熟悉就好了。


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值