mirror of
https://github.com/velocitatem/PHANTOM.git
synced 2026-05-31 08:33:36 +00:00
23 lines
698 B
Python
23 lines
698 B
Python
from sklearn.pipeline import Pipeline
|
|
from sklearn.preprocessing import StandardScaler
|
|
|
|
from extract import KafkaDataFetcher, ExperimentJoiner, EventTitleAugmenter
|
|
from mapping import SessionTransitionProbMatrixTransformer, render_graph
|
|
from demand import DemandEstimator
|
|
|
|
|
|
# exposable pipelines
|
|
etl_pipeline = Pipeline([
|
|
('kafka_fetch', KafkaDataFetcher()),
|
|
('experiment_join', ExperimentJoiner()),
|
|
('event_augment', EventTitleAugmenter()),
|
|
])
|
|
pricing_pipeline = Pipeline([
|
|
('demand_estimation', DemandEstimator()),
|
|
])
|
|
|
|
if __name__ == "__main__":
|
|
processed_data = etl_pipeline.fit_transform(None)
|
|
pricing = pricing_pipeline.fit_transform(processed_data)
|
|
print(pricing)
|