数据分析入门之KNN影片类型和癌症预测

文章目录

1、预测电影类型

一部电影的类型可以通过它的武打镜头和接吻镜头的多少了确定影片是 动作片 还是 爱情片

1.1、导入相关库

import numpy as np
import pandas as pd
from pandas import Series, DataFrame
from sklearn.neighbors import KNeighborsClassifier

1.2、导入数据

准备数据，Dataframe类型的数据，numpy类型的数据
数据 x 必须是二维的[[样本一], [样本二], [样本三]]
目标值 y 没有定义要求

data = pd.read_excel('movies.xlsx', sheet_name=1)
data

1.3、切分出 x 和 y

x = data[['武打镜头', '接吻镜头']]
y = data['分类情况']

1.4、声明算法

knn = KNeighborsClassifier(n_neighbors=5)

1.5、进行训练

knn.fit(x, y)

KNeighborsClassifier(algorithm='auto', leaf_size=30, metric='minkowski',
                     metric_params=None, n_jobs=None, n_neighbors=5, p=2,
                     weights='uniform')

1.6、生成数据(导入预测值)

1、碟中谍6——>动作片
2、李茶的姑妈——>爱情片

x_test = np.array([[100, 2], [2, 15]])

1.7、使用算法进行预测

knn.predict(x_test)

array(['动作片', '爱情片'], dtype=object)

2、预测是否患癌症

细胞核的特征:

1. 第一个特征是ID；第二个特征是这个案例的癌症诊断结果，编码“M”表示恶性，“B”表示良性。

1. 其他30个特征是数值的其他指标，包括细胞核的半径（Radius）、质地(Texture)、周长(Perimeter)、面积(Area)和光滑度(Smoothness)等的均值、标准差和最大值。

2.1、获取数据

x数据，样本（人）特征，细胞核的质地、光滑度等
y目标值，样本（人）诊断结果、两种：M恶性，B良性

2.1.1、导入数据

import numpy as np
import pandas as pd
from pandas import Series, DataFrame
from sklearn.neighbors import KNeighborsClassifier
# 导入数据
# 二分类问题，良性和恶行
data = pd.read_csv('cancer.csv', sep='\t')
data

2.1.2、切分出 x数据 和 y目标值

将数据一分为二，划分后数据是一 一对应的

# 诊断结果、目标值
y = data['Diagnosis']
# 训练数据，特征，细胞核一系列特征，光滑度，质地等
x = data.iloc[:,2:] #第二列后面的所有数据

2.1.3、分出训练集和测试集

from sklearn.model_selection import train_test_split
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2) #使用20%的数据来测试数据

2.3、声明算法并学习

knn = KNeighborsClassifier()
# 算法学习455个样本
knn.fit(x_train, y_train)

KNeighborsClassifier(algorithm='auto', leaf_size=30, metric='minkowski',
                     metric_params=None, n_jobs=None, n_neighbors=5, p=2,
                     weights='uniform')

2.4、结果预测

# 使用算法进行预测
y_pre = knn.predict(x_test)
y_pre

array(['B', 'B', 'B', 'B', 'B', 'M', 'M', 'B', 'B', 'M', 'B', 'B', 'M',
       'B', 'B', 'B', 'M', 'B', 'B', 'M', 'M', 'B', 'M', 'M', 'B', 'B',
       'B', 'B', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'B', 'M', 'M', 'M',
       'B', 'M', 'M', 'B', 'B', 'B', 'M', 'B', 'B', 'M', 'M', 'M', 'B',
       'B', 'B', 'B', 'B', 'M', 'B', 'M', 'B', 'B', 'B', 'M', 'B', 'M',
       'B', 'M', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'B', 'B', 'B', 'B',
       'M', 'B', 'M', 'B', 'M', 'B', 'M', 'B', 'B', 'B', 'B', 'B', 'M',
       'M', 'M', 'B', 'M', 'M', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'M',
       'B', 'B', 'B', 'M', 'B', 'B', 'M', 'B', 'B', 'M'], dtype=object)

2.5、概率预测

knn.predict_proba(x_test) # proba概率

array([[1. , 0. ],
       [0.8, 0.2],
       [1. , 0. ],
       [1. , 0. ],
       [1. , 0. ],
......
       [1. , 0. ],
       [0. , 1. ],
       [1. , 0. ],
       [0.6, 0.4],
       [0. , 1. ]])

2.6、预测值与真实值对比

# 真实情况进行比对，验证算法好不好
display(y_pre)
# 刚才保留，真实的数据
display(y_test.values)

array(['B', 'B', 'B', 'B', 'B', 'M', 'M', 'B', 'B', 'M', 'B', 'B', 'M',
       'B', 'B', 'B', 'M', 'B', 'B', 'M', 'M', 'B', 'M', 'M', 'B', 'B',
       'B', 'B', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'B', 'M', 'M', 'M',
       'B', 'M', 'M', 'B', 'B', 'B', 'M', 'B', 'B', 'M', 'M', 'M', 'B',
       'B', 'B', 'B', 'B', 'M', 'B', 'M', 'B', 'B', 'B', 'M', 'B', 'M',
       'B', 'M', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'B', 'B', 'B', 'B',
       'M', 'B', 'M', 'B', 'M', 'B', 'M', 'B', 'B', 'B', 'B', 'B', 'M',
       'M', 'M', 'B', 'M', 'M', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'M',
       'B', 'B', 'B', 'M', 'B', 'B', 'M', 'B', 'B', 'M'], dtype=object)
       
array(['B', 'B', 'B', 'B', 'B', 'M', 'M', 'M', 'B', 'M', 'B', 'B', 'B',
       'B', 'B', 'B', 'M', 'B', 'B', 'M', 'M', 'B', 'M', 'M', 'B', 'B',
       'B', 'B', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'B', 'M', 'M', 'M',
       'B', 'M', 'M', 'M', 'B', 'B', 'B', 'B', 'M', 'M', 'M', 'M', 'B',
       'B', 'B', 'B', 'B', 'M', 'B', 'M', 'B', 'B', 'B', 'M', 'B', 'M',
       'B', 'M', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'B', 'B', 'B', 'B',
       'M', 'B', 'M', 'B', 'M', 'B', 'M', 'B', 'B', 'B', 'B', 'B', 'M',
       'B', 'M', 'B', 'M', 'M', 'B', 'B', 'B', 'B', 'B', 'M', 'B', 'M',
       'M', 'B', 'B', 'M', 'B', 'B', 'M', 'B', 'M', 'M'], dtype=object)

2.7、计算预测的准确率

方法一：均值法

用预测值和真实值对比，返回结果为 True 和 False ，对其求均值得到准确率

(y_pre == y_test.values).mean()

0.9298245614035088

方法二：knn.score()

knn.score(x_test, y_test) #x_test是用来测试的数据， y_test是真实的数据

0.9298245614035088