def get_simple_class_pipeline(): final = PrimaryNode(operation_type='logit') pipeline = Pipeline(final) return pipeline
def get_simple_regr_pipeline(): final = PrimaryNode(operation_type='xgbreg') pipeline = Pipeline(final) return pipeline