Update train.py

train.py

@@ -17,7 +17,9 @@ def run_training(max_depth=10, min_samples_split=2):
     # Load dataset
     train_dataset = pd.read_csv(train_path)
     test_dataset = pd.read_csv(test_path)
-
+    X_train = train_dataset.drop("is_click", axis="columns")
+    y_train = train_dataset["is_click"]
+    
     
     # Define DecisionTreeClassifier and GridSearch
     param_grid = {