Skip to content

Commit 8b1f9f0

Browse files
DOC remove as_frame to avoide readthedocs install pandas (#90)
1 parent 3fddec5 commit 8b1f9f0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/plot_pruning.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,10 @@
2727
from sklearn.datasets import load_iris
2828
from sklearn.linear_model import LogisticRegression
2929

30-
iris = load_iris(as_frame=True)
30+
iris = load_iris()
3131
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]
32+
X_train = iris["data"][10:120]
33+
y_train = iris["target"][10:120]
3434

3535
# %%
3636
# Random data pruning

0 commit comments

Comments
 (0)