We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3fddec5 commit 8b1f9f0Copy full SHA for 8b1f9f0
examples/plot_pruning.py
@@ -27,10 +27,10 @@
27
from sklearn.datasets import load_iris
28
from sklearn.linear_model import LogisticRegression
29
30
-iris = load_iris(as_frame=True)
+iris = load_iris()
31
baseline_lr = LogisticRegression(max_iter=1000).fit(iris["data"], iris["target"])
32
-X_train = iris["data"].values[10:120]
33
-y_train = iris["target"].values[10:120]
+X_train = iris["data"][10:120]
+y_train = iris["target"][10:120]
34
35
# %%
36
# Random data pruning
0 commit comments