码迷,mamicode.com
首页 > 其他好文 > 详细

和scikit-learn打个招呼

时间:2018-01-11 15:52:05      阅读:127      评论:0      收藏:0      [点我收藏+]

标签:div   scipy   int   pos   rom   pre   ssi   ict   gpo   

 

1.先装对应的库。不能偷懒,都得装,不然飞不起来。

pip install scikit-learn

pip install numpy

pip install scipy

 

 

2.测试如下代码。

 1 import sklearn
 2 from sklearn import tree
 3 
 4 # 用于学习的特征数据 [重量, 表皮光滑]
 5 features = [[140, 1], [130, 1], [150, 0], [170, 0]]
 6 
 7 # 用于学习的标准答案
 8 labels = [苹果, 苹果, 梨子, 梨子]
 9 
10 # 开始学习(训练分类器)
11 clf = tree.DecisionTreeClassifier()
12 clf = clf.fit(features, labels)
13 
14 # 进行预测
15 ret = clf.predict([[150, 0]])
16 
17 print(ret)

 

输出:

[梨子]

 

做到这里,咱们和scikit-learn算是打了个招呼,混个脸熟。

 

和scikit-learn打个招呼

标签:div   scipy   int   pos   rom   pre   ssi   ict   gpo   

原文地址:https://www.cnblogs.com/hatemath/p/8267263.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!