和scikit-learn打个招呼

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

pip install scikit-learn
 
pip install numpy
 
pip install scipy

2.测试如下代码。

 import sklearn
 from sklearn import tree
 
 # 用于学习的特征数据 [重量, 表皮光滑]
 features = [[140, 1], [130, 1], [150, 0], [170, 0]]
 
 # 用于学习的标准答案
 labels = ['苹果', '苹果', '梨子', '梨子']
 
 # 开始学习（训练分类器）
 clf = tree.DecisionTreeClassifier()
 clf = clf.fit(features, labels)
 
 # 进行预测
 ret = clf.predict([[150, 0]])
 
 print(ret)

输出：

['梨子']

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