mirror of
https://github.com/velocitatem/PHANTOM.git
synced 2026-05-31 08:33:36 +00:00
21 lines
582 B
Python
21 lines
582 B
Python
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__":
|
|
processed_data = etl_pipeline.fit_transform(None)
|
|
pricing = pricing_pipeline.fit_transform(processed_data)
|