from sklearn import tree
features = [[140, 1], [130, 1], [150, 0], [170, 0]]
labels = [0,0,1,1]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(features, labels)
from colorama import init, Fore, Back, Style #Ger texten färg efter fruktens färg i detta fall Gröm för Äppelsin

init(convert=True)

print(Fore.GREEN + 'Äpple')
print(clf.predict([[145, 1]])), # [0] betyder äpple

from colorama import init, Fore, Back, Style #Ger texten färg efter fruktens färg i detta fall gul (närmast orange) för appelsin

init(convert=True)

print(Fore.YELLOW + 'Applesin')
print(clf.predict([[165, 0]])), # [1] betyder äppelsin