from collections.abc import Iterable
from typing import Any, ClassVar
from typing_extensions import Self

from numpy import ndarray
from pandas.core.frame import DataFrame
from scipy.sparse import csr_matrix, spmatrix

from .._typing import ArrayLike, Float, Int, MatrixLike
from ..base import BaseEstimator, TransformerMixin

# Authors: Alexandre Gramfort <alexandre.gramfort@inria.fr>
#          Mathieu Blondel <mathieu@mblondel.org>
#          Olivier Grisel <olivier.grisel@ensta.org>
#          Andreas Mueller <amueller@ais.uni-bonn.de>
#          Joel Nothman <joel.nothman@gmail.com>
#          Hamzeh Alsalhi <ha258@cornell.edu>
# License: BSD 3 clause

__all__ = [
    "label_binarize",
    "LabelBinarizer",
    "LabelEncoder",
    "MultiLabelBinarizer",
]

class LabelEncoder(TransformerMixin, BaseEstimator):
    classes_: ndarray = ...

    def fit(self, y: ArrayLike) -> Self: ...
    def fit_transform(self, y: ArrayLike) -> ArrayLike: ...
    def transform(self, y: ArrayLike) -> ArrayLike: ...
    def inverse_transform(self, y: ArrayLike) -> ndarray: ...

class LabelBinarizer(TransformerMixin, BaseEstimator):
    sparse_input_: bool = ...
    y_type_: str = ...
    classes_: ndarray = ...

    _parameter_constraints: ClassVar[dict] = ...

    def __init__(self, *, neg_label: Int = 0, pos_label: Int = 1, sparse_output: bool = False) -> None: ...
    def fit(self, y: MatrixLike | ArrayLike) -> Self: ...
    def fit_transform(self, y: MatrixLike | ArrayLike) -> ndarray | spmatrix: ...
    def transform(self, y: MatrixLike | ArrayLike) -> ndarray | spmatrix: ...
    def inverse_transform(self, Y: MatrixLike, threshold: None | Float = None) -> ndarray | spmatrix: ...

def label_binarize(
    y: ArrayLike | DataFrame, *, classes: ArrayLike, neg_label: Int = 0, pos_label: Int = 1, sparse_output: bool = False
) -> ndarray | spmatrix: ...

class MultiLabelBinarizer(TransformerMixin, BaseEstimator):
    classes_: ndarray = ...

    _parameter_constraints: ClassVar[dict] = ...

    def __init__(self, *, classes: None | ArrayLike = None, sparse_output: bool = False) -> None: ...
    def fit(self, y: Iterable[Iterable] | list[range]) -> Self: ...
    def fit_transform(self, y: Iterable[Iterable]) -> ndarray | spmatrix: ...
    def transform(self, y: list[list[Any | Int]] | Iterable[Iterable]) -> csr_matrix | ndarray: ...
    def inverse_transform(self, yt: MatrixLike) -> ndarray: ...