coffeine.covariance_transformers.ExpandFeatures

class coffeine.covariance_transformers.ExpandFeatures(estimator: Pipeline, expander_column: str)

Add binary interaction terms after projection step.

Simple ad-hoc interaction features in projected space by multiplying a scaler (continuous or categorical) sample-level feature with the representation obtained from projection and vectorization, e.g., drug dosage or biomarker value at baseline.

Parameters:

estimatorsklearn pipeline

A coffeine filter-bank transformer, regressor or classifier.

expander_columnstr

The column in the coffeine data frame (passed through as reminder) that should be used for computing the interaction features by multiplication.