21 lines
750 B
Python
Executable File
21 lines
750 B
Python
Executable File
#!/usr/bin/env python
|
|
from sklearn import set_config
|
|
from sklearn.pipeline import make_pipeline
|
|
from sklearn.preprocessing import OneHotEncoder, StandardScaler
|
|
from sklearn.impute import SimpleImputer
|
|
from sklearn.compose import make_column_transformer
|
|
from sklearn.linear_model import LogisticRegression
|
|
set_config(display='diagram')
|
|
|
|
num_proc = make_pipeline(SimpleImputer(strategy='median'), StandardScaler())
|
|
|
|
cat_proc = make_pipeline(
|
|
SimpleImputer(strategy='constant', fill_value='missing'),
|
|
OneHotEncoder(handle_unknown='ignore'))
|
|
|
|
preprocessor = make_column_transformer((num_proc, ('feat1', 'feat3')),
|
|
(cat_proc, ('feat0', 'feat2')))
|
|
|
|
clf = make_pipeline(preprocessor, LogisticRegression())
|
|
clf
|