机器学习---预测电影类型

本文主要是介绍机器学习---预测电影类型，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

#coding=UTF8 
from numpy import *
import operatordef createDataSet():group = array([[3,104],[2,100],[1,81],[101,10],[99,5]]) #训练集labels = ['affectional film','affectional film','affectional film','action movie',"action movie"]return group,labelsdef classify0(inX,dataSet,labels,k):dataSetSize = dataSet.shape[0]diffMat = tile(inX, (dataSetSize,1)) - dataSetsqDiffMat = diffMat ** 2; #diffMat ^ 2sqDistances = sqDiffMat.sum(axis=1) #将矩阵的每一行相加比如[[2,1,3],[1,1,1]]结果为[6,3]distances = sqDistances ** 0.5 #sqDistances ^ (1/2)sortedDistIndicies = distances.argsort()classCount = {}for i in range(k):voteIlabel = labels[sortedDistIndicies[i]]classCount[voteIlabel] = classCount.get(voteIlabel,0) + 1sortedClassCount = sorted(classCount.iteritems(),key=operator.itemgetter(1),reverse=True)   #获取对象的第1个域的值return sortedClassCount[0][0]group,labels = createDataSet()
print("输入武打镜头的个数:")
fun1 = input() #武打镜头的个数
print("输入亲吻镜头的个数:")
fun2 = input() #亲吻镜头的个数
b = classify0([fun1,fun2], group, labels, 3) #[7,100],7代表武打镜头为7,100代表接吻镜头为100,根据训练集可以得出这是一部爱情片
if b == "affectional film" :print("这很可能是一部爱情片")
else:print("这很可能是一部动作片")

这篇关于机器学习---预测电影类型的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---