xiaowei1234 · November 8, 2018 00:24
diff --git a/sklearn2pmml_pipe.py b/sklearn2pmml_pipe.py
 pl = PMMLPipeline([
    ('featureUnion', featureU)
    , ('impute', Imputer(strategy='median'))
    , ('standardize', StandardScaler())
 #     , ('interactions', PolynomialFeatures(include_bias=False))
 #     , ('clf', SGDClassifier(alpha=0.008, l1_ratio=0.13, max_iter=450,loss='log'
 #                             ,penalty='elasticnet', n_iter=None, tol=None))# alpha = 0.8
    , ('clf', LogisticRegression(penalty='l2', max_iter=500, C=0.8))
 ])

 pl.fit(X, y)
 sklearn2pmml(pl, '../outputs/pmml_file.pmml', with_repr=True)
	pl = PMMLPipeline([
	('featureUnion', featureU)
	, ('impute', Imputer(strategy='median'))
	, ('standardize', StandardScaler())
	# , ('interactions', PolynomialFeatures(include_bias=False))
	# , ('clf', SGDClassifier(alpha=0.008, l1_ratio=0.13, max_iter=450,loss='log'
	# ,penalty='elasticnet', n_iter=None, tol=None))# alpha = 0.8
	, ('clf', LogisticRegression(penalty='l2', max_iter=500, C=0.8))
	])

	pl.fit(X, y)
	sklearn2pmml(pl, '../outputs/pmml_file.pmml', with_repr=True)