Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 9 additions & 6 deletions hiclass/HierarchicalClassifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,15 +11,18 @@
from joblib import Parallel, delayed
from sklearn.base import BaseEstimator
from sklearn.linear_model import LogisticRegression
from sklearn.utils.validation import _check_sample_weight
from sklearn.utils.validation import check_array, check_is_fitted
from sklearn.utils.validation import (
_check_sample_weight,
check_array,
check_is_fitted,
validate_data,
)

from hiclass.probability_combiner import (
GeometricMeanCombiner,
ArithmeticMeanCombiner,
GeometricMeanCombiner,
MultiplyCombiner,
)

from hiclass.probability_combiner import (
init_strings as probability_combiner_init_strings,
)
Expand Down Expand Up @@ -173,8 +176,8 @@ def _pre_fit(self, X, y, sample_weight):
# Check that X and y have correct shape
# and convert them to np.ndarray if need be

self.X_, self.y_ = self._validate_data(
X, y, multi_output=True, accept_sparse="csr", allow_nd=True
self.X_, self.y_ = validate_data(
self, X, y, multi_output=True, accept_sparse="csr", allow_nd=True
)

if sample_weight is not None:
Expand Down