Source code for sklearn_utilities.utils

from .types import TXPandas, TYPandas



[docs]
def intersect_X_y(X: TXPandas, y: TYPandas) -> tuple[TXPandas, TYPandas]:
    idx = X.index.intersection(y.index)
    return X.loc[idx], y.loc[idx]




[docs]
def drop_X_y(X: TXPandas, y: TYPandas) -> tuple[TXPandas, TYPandas]:
    return intersect_X_y(X, y.dropna())