Skip to content

Commit

Permalink
[squashme] wip
Browse files Browse the repository at this point in the history
  • Loading branch information
augustebaum committed Oct 4, 2024
1 parent c0661a4 commit 5e5575e
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions tests/integration/test_cross_validate.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
from skore.cross_validate import cross_validate


def test_cross_validate(in_memory_project):
from sklearn import datasets, linear_model

diabetes = datasets.load_diabetes()
X = diabetes.data[:150]
y = diabetes.target[:150]
lasso = linear_model.Lasso()

in_memory_project.put(
"my_cross_val", cross_validate(lasso, X, y, cv=3), on_error="raise"
)
breakpoint()

0 comments on commit 5e5575e

Please sign in to comment.