Skip to content

Scikit-learn

Cheatsheet- Scikit-Learn & Caret Package for Python & R respectively

Modern API Features

  • Pipeline: make_pipeline for easier composition.
  • ColumnTransformer: For applying transformations to different column types.
  • HistGradientBoostingClassifier: Fast gradient boosting.
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import StandardScaler, OneHotEncoder
from sklearn.compose import ColumnTransformer
from sklearn.ensemble import HistGradientBoostingClassifier

# Pipeline
clf = make_pipeline(StandardScaler(), HistGradientBoostingClassifier())

# Column Transformer
preprocessor = ColumnTransformer(
    transformers=[
        ("num", StandardScaler(), ["numerical_col"]),
        ("cat", OneHotEncoder(), ["categorical_col"]),
    ]
)