velocitatem/PHANTOM
1
0
Fork 0
mirror of https://github.com/velocitatem/PHANTOM.git synced 2026-05-31 08:33:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor pipeline refactor

Browse Source
This commit is contained in:
Daniel Rosel
2025-11-22 21:09:04 +01:00
parent 3cb1201acc
commit 2661b841fc
1 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
experiments/procesing/pipeline.py
View File

@@ -1,15 +1,20 @@
from sklearn.pipeline import Pipeline from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler from sklearn.preprocessing import StandardScaler
from extract import DataExtractor from extract import DataExtractor
from mapping import SessionTransitionProbMatrixTransformer, render_graph 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__": if __name__ == "__main__":
steps = [ processed_data = etl_pipeline.fit_transform(None)
('data_extraction', DataExtractor()), pricing = pricing_pipeline.fit_transform(processed_data)
#('transition_matrix', SessionTransitionProbMatrixTransformer(threshold=0.05)),
]
pipeline = Pipeline(steps)
result = pipeline.fit_transform(None)
print(result)
print(result.info())