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

migrating pricers

Browse Source
This commit is contained in:
Daniel Rosel
2025-11-28 17:38:25 +01:00
parent 73e46200c7
commit 8fae7851a6
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
experiments/procesing/steps/pricing.py
View File

@@ -1,7 +1,7 @@
import numpy as np import numpy as np
import pandas as pd import pandas as pd
from procesing.steps.base import BaseContextStep from procesing.steps.base import BaseContextStep
from procesing.pricing import ElasticityBasedPricingFunction from procesing.pricers import ElasticityBasedPricer
class StateSpace: class StateSpace:
"""State representation for pricing functions""" """State representation for pricing functions"""
@@ -57,7 +57,7 @@ class FitPricingFunctionStep(BaseContextStep):
""" """
def transform(self, elasticity_df: pd.DataFrame): def transform(self, elasticity_df: pd.DataFrame):
pricing_class = self.context.config.get('pricing_function_class', ElasticityBasedPricingFunction) pricing_class = self.context.config.get('pricing_function_class', ElasticityBasedPricer)
pricing_params = self.context.config.get('pricing_function_params', {}) pricing_params = self.context.config.get('pricing_function_params', {})
pricer = pricing_class(**pricing_params) pricer = pricing_class(**pricing_params)
@@ -79,7 +79,7 @@ class PredictPricesStep(BaseContextStep):
products = self.context.products products = self.context.products
product_ids = products['id'].values product_ids = products['id'].values
predicted_prices = pricer.transform(state_space, product_ids) predicted_prices = pricer.predict(state_space)
return pd.DataFrame({ return pd.DataFrame({
'productId': product_ids, 'productId': product_ids,