We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 090ece0 commit 297349dCopy full SHA for 297349d
skorch/exceptions.py
@@ -1,11 +1,13 @@
1
"""Contains skorch-specific exceptions and warnings."""
2
3
+from sklearn.exceptions import NotFittedError
4
+
5
6
class SkorchException(BaseException):
7
"""Base skorch exception."""
8
9
-class NotInitializedError(SkorchException):
10
+class NotInitializedError(SkorchException, NotFittedError):
11
"""Module is not initialized, please call the ``.initialize``
12
method or train the model by calling ``.fit(...)``.
13
0 commit comments