使用Tensorflow自定义一个线性分类器对‘良/恶性乳腺癌肿瘤’进行预测

1 Tensorflow作为一个开源框架，在深度学习与机器学习方面有着很大的应用。

   对于Tensorflw就不作介绍，仅仅对其应用简单的实现一下

利用tensorflow自定义一个线性分类器对‘良/恶性乳腺癌肿瘤’进行预测

2 代码实现及结果截图

#coding:utf-8
#使用tensorflow自定义一个线性分类器对‘良/恶性乳腺癌肿瘤’进行预测

import tensorflow as tf
import numpy as np
import pandas as pd
train=pd.read_csv('/home/guo/桌面/Work/shuju/Breast-Cancer/breast-cancer-train.csv')
test=pd.read_csv('/home/guo/桌面/Work/shuju/Breast-Cancer/breast-cancer-test.csv')
x_train=np.float32(train[['Clump Thickness','Cell Size']].T)
y_train=np.float32(train['Type'].T)
x_test=np.float32(test[['Clump Thickness','Cell Size']].T)
y_test=np.float32(test['Type'].T)

#定义模型
b=tf.Variable(tf.zeros([1]))
w=tf.Variable(tf.random_uniform([1,2],-1,.0,1.0))
y=tf.matmul(w, x_train)+b
loss=tf.reduce_mean(tf.square(y-y_train))
optimizer=tf.train.GradientDescentOptimizer(0.01)
train=optimizer.minimize(loss)
init=tf.initialize_all_variables()
sess=tf.Session()
sess.run(init)
for step in xrange(0,1000):
    sess.run(train)
    if step%200==0:
        print step,sess.run(w),sess.run(b)


test_negative=test.loc[test['Type']==0][['Clump Thickness','Cell Size']]
test_positive=test.loc[test['Type']==0][['Clump Thickness','Cell Size']]
import matplotlib.pyplot as plt
plt.scatter(test_negative['Clump Thickness'], test_negative['Cell Size'],marker='o',s=200,c='red')
plt.scatter(test_positive['Clump Thickness'], test_positive['Cell Size'],marker='x',s=150,c='black')
plt.xlabel('Clump Thickness')
plt.ylabel('Cell Size')
lx=np.arange(0,12)
ly=(0.5-sess.run(b)-lx*sess.run(w)[0][0])/sess.run(w)[0][1]
plt.plot(lx,ly,color='green')

plt.show()