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

05-11 19:28