博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
tensorflow 线性回归解决 iris 2分类
阅读量:7110 次
发布时间:2019-06-28

本文共 3249 字,大约阅读时间需要 10 分钟。

 

# Combining Everything Together#----------------------------------# This file will perform binary classification on the# iris dataset. We will only predict if a flower is# I.setosa or not.## We will create a simple binary classifier by creating a line# and running everything through a sigmoid to get a binary predictor.# The two features we will use are pedal length and pedal width.## We will use batch training, but this can be easily# adapted to stochastic training.import matplotlib.pyplot as pltimport numpy as npfrom sklearn import datasetsimport tensorflow as tffrom tensorflow.python.framework import opsops.reset_default_graph()# Load the iris data# iris.target = {0, 1, 2}, where '0' is setosa# iris.data ~ [sepal.width, sepal.length, pedal.width, pedal.length]iris = datasets.load_iris()binary_target = np.array([1. if x==0 else 0. for x in iris.target])iris_2d = np.array([[x[2], x[3]] for x in iris.data])# Declare batch sizebatch_size = 20# Create graphsess = tf.Session()# Declare placeholdersx1_data = tf.placeholder(shape=[None, 1], dtype=tf.float32)x2_data = tf.placeholder(shape=[None, 1], dtype=tf.float32)y_target = tf.placeholder(shape=[None, 1], dtype=tf.float32)# Create variables A and b (0 = x1 - A*x2 + b)A = tf.Variable(tf.random_normal(shape=[1, 1]))b = tf.Variable(tf.random_normal(shape=[1, 1]))# Add model to graph:# x1 - A*x2 + bmy_mult = tf.matmul(x2_data, A)my_add = tf.add(my_mult, b)my_output = tf.subtract(x1_data, my_add)# Add classification loss (cross entropy)xentropy = tf.nn.sigmoid_cross_entropy_with_logits(logits=my_output, labels=y_target)# Create Optimizermy_opt = tf.train.GradientDescentOptimizer(0.05)train_step = my_opt.minimize(xentropy)# Initialize variablesinit = tf.global_variables_initializer()sess.run(init)# Run Loopfor i in range(1000):    rand_index = np.random.choice(len(iris_2d), size=batch_size)    #rand_x = np.transpose([iris_2d[rand_index]])    rand_x = iris_2d[rand_index]    rand_x1 = np.array([[x[0]] for x in rand_x])    rand_x2 = np.array([[x[1]] for x in rand_x])    #rand_y = np.transpose([binary_target[rand_index]])    rand_y = np.array([[y] for y in binary_target[rand_index]])    sess.run(train_step, feed_dict={x1_data: rand_x1, x2_data: rand_x2, y_target: rand_y})    if (i+1)%200==0:        print('Step #' + str(i+1) + ' A = ' + str(sess.run(A)) + ', b = ' + str(sess.run(b)))        # Visualize Results# Pull out slope/intercept[[slope]] = sess.run(A)[[intercept]] = sess.run(b)# Create fitted linex = np.linspace(0, 3, num=50)ablineValues = []for i in x:  ablineValues.append(slope*i+intercept)# Plot the fitted line over the datasetosa_x = [a[1] for i,a in enumerate(iris_2d) if binary_target[i]==1]setosa_y = [a[0] for i,a in enumerate(iris_2d) if binary_target[i]==1]non_setosa_x = [a[1] for i,a in enumerate(iris_2d) if binary_target[i]==0]non_setosa_y = [a[0] for i,a in enumerate(iris_2d) if binary_target[i]==0]plt.plot(setosa_x, setosa_y, 'rx', ms=10, mew=2, label='setosa')plt.plot(non_setosa_x, non_setosa_y, 'ro', label='Non-setosa')plt.plot(x, ablineValues, 'b-')plt.xlim([0.0, 2.7])plt.ylim([0.0, 7.1])plt.suptitle('Linear Separator For I.setosa', fontsize=20)plt.xlabel('Petal Length')plt.ylabel('Petal Width')plt.legend(loc='lower right')plt.show()

 

转载地址:http://vqlhl.baihongyu.com/

你可能感兴趣的文章
SVG和Vector的概念和如何在Android Studio中使用
查看>>
Swift练习题—基础控制流
查看>>
技术专栏丨从原理到应用,Elasticsearch详解(上)
查看>>
什么是散列表(Hash Table)
查看>>
vue作用域插槽,你真的懂了吗?
查看>>
透视云原生热的背后
查看>>
个人整合,java 通过aspose转PDF ,支持各种格式 JPG ,TXT, PPT, EXCEL, DOC 免费开箱即用版...
查看>>
如果使用Github管理代码的方式文章
查看>>
菜鸟成长之路 第二周
查看>>
麻省理工教授透露为什么80%黑客都使用Python!
查看>>
linux dhcp服务器 超级作用域
查看>>
二分查找
查看>>
对haproxy配置学习过程中几个点进行总结
查看>>
Oracle资源配置profile(二,2/2)
查看>>
IntelliJ IDEA 12 详细开发教程(二)Tomcat服务配置与Jrebel热部署
查看>>
phpadmin 详细配置
查看>>
Cisco IOS 配置PPPOE
查看>>
PHP: 深入了解一致性哈希
查看>>
outlook 2003配置失败:到服务器的连接不可达
查看>>
SQLServer 常用监控性能DMV & DMF
查看>>