Skip to content
Snippets Groups Projects
Commit 4b6a46a9 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[tests] Fix use of classification evaluator

parent 661ab557
No related branches found
No related tags found
No related merge requests found
Pipeline #95349 passed
...@@ -40,10 +40,13 @@ def test_run_binary_1() -> None: ...@@ -40,10 +40,13 @@ def test_run_binary_1() -> None:
("s3", [1], [0.4]), ("s3", [1], [0.4]),
] ]
rng = numpy.random.default_rng(42)
results = run( results = run(
"test", "test",
predictions, predictions,
binning=10, binning=10,
rng=rng,
threshold_a_priori=0.5, threshold_a_priori=0.5,
) )
...@@ -97,11 +100,14 @@ def test_run_binary_2() -> None: ...@@ -97,11 +100,14 @@ def test_run_binary_2() -> None:
("s3", [1], [0.4]), ("s3", [1], [0.4]),
] ]
rng = numpy.random.default_rng(42)
# a change in the threshold should not affect auc and average precision scores # a change in the threshold should not affect auc and average precision scores
results = run( results = run(
"test", "test",
predictions, predictions,
binning=10, binning=10,
rng=rng,
threshold_a_priori=0.3, threshold_a_priori=0.3,
) )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment