Sklearn决策树分类器-动物猜谜游戏

debugcn 发表于 Dev

用户13784970

我正在尝试使用sklearn的决策树分类器制作动物猜谜游戏。这样，在用户输入的情况下，它将说出动物是蜘蛛还是鱼。当我输入腿的数量及其停留的位置时，它说动物是鱼，反之亦然。这是代码，有什么想法吗？

from sklearn.tree import DecisionTreeClassifier

clf = DecisionTreeClassifier()
features=[[6,0],
      [6,0],                #0 is spider and 1 is fish
      [0,1],                #6 = 4 legs and 8=stays in land ( this is a spider)
      [0,1]]               #4=no legs and 7 is stays in water (this is a fish)
     

outcomes=[1,1,0,0]

clf.fit(features, outcomes)

legs=int(input("Enter the number of legs your animal has: "))
land_water=input("Enter whether you animal stays in land or water: ")
if land_water=="land":
   land_water=0
else:
   land_water=1

sa=clf.predict([[legs, land_water]])

if sa==[0]:
    print("Animal is a spider")
else:
    print('Animal is a fish')

戴维·布雷克斯

您反转了outcomes，这样对我有用：

from sklearn.tree import DecisionTreeClassifier

clf = DecisionTreeClassifier()
features=[[6,0],
      [6,0],                #0 is spider and 1 is fish
      [0,1],                #6 = 4 legs and 8=stays in land ( this is a spider)
      [0,1]]               #4=no legs and 7 is stays in water (this is a fish)
     

outcomes=["spider","spider","fish","fish"]

clf.fit(features, outcomes)

legs=int(input("Enter the number of legs your animal has: "))
land_water=input("Enter whether you animal stays in land or water: ")
if land_water=="land":
   land_water=0
else:
   land_water=1

sa=clf.predict([[legs, land_water]])
print(sa)
if sa[0] == "spider":
    print("Animal is a spider")
else:
    print('Animal is a fish')

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。