Tầm quan trọng của đặc trưng

import pandas as pd
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier

depth = int(input())

# Mỗi hàng ứng với một người mới học: 1 trong standing nghĩa là họ đã đứng lên ván trong tuần đầu tiên
surf = pd.read_csv("surfschool.csv")
features = ["age", "fitness", "days_surfed", "skated_before", "lesson_hours", "wetsuit_fit"]
X_train, X_test, y_train, y_test = train_test_split(surf[features], surf["standing"], test_size=0.25, random_state=0)
tree = DecisionTreeClassifier(max_depth=depth, random_state=0)
tree.fit(X_train, y_train)

fig, ax = plt.subplots(layout="constrained")  # chừa chỗ cho các tên dài
ax.set_title("What Gets Beginners Standing?")
ax.set_xlabel("Importance")

# Viết code của bạn bên dưới