和scikit-learn打个招呼

Posted 2020-10-20 hatemath

 

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算是打了个招呼，混个脸熟。