Update app.py

app.py

@@ -106,7 +106,7 @@ def do_train(n_samples, n_new_data):
     probable_clusters = inductive_learner.predict(X_new)
     fig3, axes3 = plt.subplots()
     disp = DecisionBoundaryDisplay.from_estimator(
-        inductive_learner, X_all, response_method="predict", alpha=0.4, ax=axes3
+        inductive_learner, X, response_method="predict", alpha=0.4, ax=axes3
     )
     disp.ax_.set_title("Classify unknown instances with known clusters")
     disp.ax_.scatter(X[:, 0], X[:, 1], c=cluster_labels, alpha=0.5, edgecolor="k")
@@ -120,7 +120,7 @@ def do_train(n_samples, n_new_data):
     classifier = RandomForestClassifier(random_state=RANDOM_STATE).fit(X_all, y)
     fig4, axes4 = plt.subplots()
     disp = DecisionBoundaryDisplay.from_estimator(
-            classifier, X, response_method="predict", alpha=0.4, ax=axes4
+            classifier, X_all, response_method="predict", alpha=0.4, ax=axes4
         )
     disp.ax_.set_title("Classify unknown instance with recomputing clusters")
     disp.ax_.scatter(X_all[:, 0], X_all[:, 1], c=y, alpha=0.5, edgecolor="k")