We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 5546553 + b7606bd commit 5d0ba0bCopy full SHA for 5d0ba0b
skorch/callbacks/scoring.py
@@ -2,12 +2,18 @@
2
3
from contextlib import contextmanager
4
from contextlib import suppress
5
+from distutils.version import LooseVersion
6
from functools import partial
7
import warnings
8
9
import numpy as np
-from sklearn.metrics.scorer import (
10
- check_scoring, _BaseScorer, make_scorer)
+import sklearn
11
+from sklearn.metrics import make_scorer, check_scoring
12
+
13
+if LooseVersion(sklearn.__version__) >= '0.22':
14
+ from sklearn.metrics._scorer import _BaseScorer
15
+else:
16
+ from sklearn.metrics.scorer import _BaseScorer
17
18
from skorch.utils import data_from_dataset
19
from skorch.utils import is_skorch_dataset
0 commit comments