SelectType¶
Selects features based on their type.
This is practical when you want to apply different preprocessing steps to different kinds of features. For instance, a common usecase is to apply a preprocessing.StandardScaler
to numeric features and a preprocessing.OneHotEncoder
to categorical features.
Parameters¶
-
types
Type → tuple[type]
Python types which you want to select. Under the hood, the
isinstance
method will be used to check if a value is of a given type.
Examples¶
import numbers
from river import compose
from river import linear_model
from river import preprocessing
num = compose.SelectType(numbers.Number) | preprocessing.StandardScaler()
cat = compose.SelectType(str) | preprocessing.OneHotEncoder()
model = (num + cat) | linear_model.LogisticRegression()
Methods¶
learn_one
Update with a set of features x
.
A lot of transformers don't actually have to do anything during the learn_one
step because they are stateless. For this reason the default behavior of this function is to do nothing. Transformers that however do something during the learn_one
can override this method.
Parameters
- x — 'dict'
transform_one
Transform a set of features x
.
Parameters
- x — 'dict'
Returns
dict: The transformed values.