Use XGBoost to select features

ashwin153 / autofeat

automated exploratory data analysis

0 stars 0 forks source link

Use XGBoost to select features #17

Closed ashwin153 closed 2 months ago

ashwin153 commented 2 months ago

https://www.rhosignal.com/posts/polars-arrow-xgboost/

ashwin153 commented 2 months ago

@dataclasses.dataclass(frozen=True, kw_only=True)
class SHAP:
    """"""

    estimator: sklearn.base.BaseEstimator
    explainer: shap.Explainer

    def score(
        self,
        df: polars.DataFrame,
    ) -> polars.DataFrame:

ashwin153 commented 2 months ago

replaced by #27