机器学习-1
来自b站的一个视频:
机器学习1
题目大意
给出一些水果的数据(苹果和橘子),数据中包含了重量和表皮光滑度
要求生成classifier(分类器)来预测输入的数据是苹果还是橘子
python代码:
# coding=utf-8
# 引入sklearn包
from sklearn import tree
# 判断是苹果还说橘子
# 数据格式:[重量(单位:克),表皮光滑]
# 0表示光滑,1表示粗糙
features = [[140,0],[130,0],[150,1],[170,1]];
# 0表示苹果,1表示橘子
labels = [0,0,1,1];
# 使用决策树过滤器
clf = tree.DecisionTreeClassifier()
# 通过数据训练
clf = clf.fit(features,labels)
# 预测一个重量为150克的,外表光滑的水果说苹果还是橘子
print clf.predict([[150,0]])