基于python的贝叶斯分类算法(数据集为Iris_data)
#coding:utf-8 import math Iris_setosa_data=[] Iris_versicolor_data=[] Iris_virginica_data=[] #读取训练数据集,这里我将每种花取前45条数据,剩下的5条数据另外存入测试数据集 def read_train_data(filename): f=open(filename,'r') all_lines=f.readlines() for line in all_lines[0:45]: line=line.strip().split(',') Iris_setosa_data.append(line[0:4]) #Iris_setosa_label+=1 for line in all_lines[51:95]: line=line.strip().split(',') Iris_versicolor_data.append(line[0:4]) #Iris_versicolor_label+=1 for line in all_lines[101:145]: line=line.strip().split(',') Iris_virginica_data.append(line[0:4]) #Iris_virginica_label+=1 return Iris_setosa_data