חשיבות התכונות

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())

# שורה אחת לכל מתחיל: 1 ב-standing פירושו שהצליח לעמוד על הגלשן במהלך השבוע הראשון שלו
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")  # משאיר מקום לשמות ארוכים
ax.set_title("What Gets Beginners Standing?")
ax.set_xlabel("Importance")

# כתבו את הקוד שלכם למטה