diff --git a/experiments/procesing/pipeline.py b/experiments/procesing/pipeline.py index 54aae61..0465317 100644 --- a/experiments/procesing/pipeline.py +++ b/experiments/procesing/pipeline.py @@ -1,15 +1,20 @@ from sklearn.pipeline import Pipeline from sklearn.preprocessing import StandardScaler + from extract import DataExtractor from mapping import SessionTransitionProbMatrixTransformer, render_graph +from demand import DemandEstimator +# exposable pipelines +etl_pipeline = Pipeline([ + ('data_extraction', DataExtractor()), +]) +pricing_pipeline = Pipeline([ + ('demand_estimation', DemandEstimator()), + ('scaling', StandardScaler()), +]) + if __name__ == "__main__": - steps = [ - ('data_extraction', DataExtractor()), - #('transition_matrix', SessionTransitionProbMatrixTransformer(threshold=0.05)), - ] - pipeline = Pipeline(steps) - result = pipeline.fit_transform(None) - print(result) - print(result.info()) + processed_data = etl_pipeline.fit_transform(None) + pricing = pricing_pipeline.fit_transform(processed_data)