Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -120,7 +120,7 @@ def do_train(n_samples, n_new_data):
|
|
120 |
classifier = RandomForestClassifier(random_state=RANDOM_STATE).fit(X_all, y)
|
121 |
fig4, axes4 = plt.subplots()
|
122 |
disp = DecisionBoundaryDisplay.from_estimator(
|
123 |
-
classifier,
|
124 |
)
|
125 |
disp.ax_.set_title("Classify unknown instance with recomputing clusters")
|
126 |
disp.ax_.scatter(X_all[:, 0], X_all[:, 1], c=y, alpha=0.5, edgecolor="k")
|
|
|
120 |
classifier = RandomForestClassifier(random_state=RANDOM_STATE).fit(X_all, y)
|
121 |
fig4, axes4 = plt.subplots()
|
122 |
disp = DecisionBoundaryDisplay.from_estimator(
|
123 |
+
classifier, X, response_method="predict", alpha=0.4, ax=axes4
|
124 |
)
|
125 |
disp.ax_.set_title("Classify unknown instance with recomputing clusters")
|
126 |
disp.ax_.scatter(X_all[:, 0], X_all[:, 1], c=y, alpha=0.5, edgecolor="k")
|