pyepo.twostage

Two-stage predict then optimize model

Submodules

• pyepo.twostage.autosklearnpred
• pyepo.twostage.sklearnpred

Functions

sklearnPred(pmodel)	Two-stage prediction and optimization with scikit-learn.
autoSklearnPred(optmodel, seed, timelimit[, metric])	Two-stage prediction and optimization with auto-sklearn.

Package Contents

pyepo.twostage.sklearnPred(pmodel)

Two-stage prediction and optimization with scikit-learn.

Parameters:

pmodel (Regressor) – scikit-learn regression model

Returns:

scikit-learn multi-output regression model

Return type:

MultiOutputRegressor

pyepo.twostage.autoSklearnPred(optmodel, seed, timelimit, metric='mse')

Two-stage prediction and optimization with auto-sklearn.

Parameters:

optmodel (optModel) – optimization model

Returns:

Auto-SKlearn multi-output regression model

Return type:

AutoSklearnRegressor