【发布时间】:2020-06-27 17:00:43
【问题描述】:
我想知道是否可以有一个包含强制元素和可选元素的管道。可选的是有条件的。例如,您可以拥有一个带有下采样元素的管道,或者您可以拥有相同的管道而不进行下采样。所以
from imblearn.pipeline import Pipeline as IMBPipeline
import xgboost as xgb
from imblearn.under_sampling import RandomUnderSampler
pipe = IMBPipeline([
('sampling',RandomUnderSampler()),
('clf', xgb.XGBClassifier(**params, n_jobs=-1))
])
例如,如果您有不平衡的数据集,则只有采样部分。但采样部分仍在进行中,只是有条件的。有这样的吗?
【问题讨论】:
标签: python scikit-learn pipeline imblearn