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