Đọc cây

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier, export_text

# Mỗi hàng ứng với một loại phô mai: hàm lượng nước (%), số tuần ủ và loại phô mai
cheeses = pd.read_csv("cheeses.csv")
features = ["moisture_pct", "aged_weeks"]
X_train, X_test, y_train, y_test = train_test_split(cheeses[features], cheeses["cheese"], test_size=0.25, random_state=0)
tree = DecisionTreeClassifier(max_depth=2, random_state=0)
tree.fit(X_train, y_train)

# Một loại phô mai mới cần xác định tên
moisture = int(input())
aged = int(input())
new_cheese = pd.DataFrame([[moisture, aged]], columns=features)

# Viết code bên dưới: in ra cây quyết định, rồi trả lời trong by_hand giống như cây


def by_hand(moisture, aged):
    return "fresh"


# Đừng thay đổi các dòng bên dưới
print("By hand:", by_hand(moisture, aged))
print("The tree:", tree.predict(new_cheese)[0])