天天看点

决策树之隐形眼镜分类

决策树之隐形眼镜分类

from sklearn import tree

if __name__ == '__main__':
    fr = open('lenses.txt')
    lenses = [inst.strip().split('\t') for inst in fr.readlines()]
    print(lenses)
    lensesLabels = ['age', 'prescript', 'astigmatic', 'tearRate']
    clf = tree.DecisionTreeClassifier()
    lenses = clf.fit(lenses, lensesLabels)      
young myope no  reduced no lenses
young myope no  normal  soft
young myope yes reduced no lenses
young myope yes normal  hard
young hyper no  reduced no lenses
young hyper no  normal  soft
young hyper yes reduced no lenses
young hyper yes normal  hard
pre myope no  reduced no lenses
pre myope no  normal  soft
pre myope yes reduced no lenses
pre myope yes normal  hard
pre hyper no  reduced no lenses
pre hyper no  normal  soft
pre hyper yes reduced no lenses
pre hyper yes normal  no lenses
presbyopic  myope no  reduced no lenses
presbyopic  myope no  normal  no lenses
presbyopic  myope yes reduced no lenses
presbyopic  myope yes normal  hard
presbyopic  hyper no  reduced no lenses
presbyopic  hyper no  normal  soft
presbyopic  hyper yes reduced no lenses
presbyopic  hyper yes normal  no lenses