Commit e4b6f3e9 authored by Barthelet Thibault's avatar Barthelet Thibault
Browse files

working yayy

parent 7fc9d59a

forecast_app/app.py

+2 −2
Original line number Diff line number Diff line
@@ -26,8 +26,8 @@ app.config['SESSION_TYPE'] = 'filesystem' 


# Load dataset

DATA_DIR = os.path.join(os.path.dirname(__file__), 'data')

TRAINING_DATA_PATH = os.path.join(DATA_DIR, 'sim_supply_chain_data_training.json')

TEST_DATA_PATH = os.path.join(DATA_DIR, 'sim_supply_chain_data_test.json')

TRAINING_DATA_PATH = os.path.join(DATA_DIR, 'sim2_supply_chain_data_training.json')

TEST_DATA_PATH = os.path.join(DATA_DIR, 'sim2_supply_chain_data_test.json')





def load_training_data():

forecast_app/smart_data_generator.py

+6 −6
Original line number Diff line number Diff line
@@ -159,7 +159,7 @@ class SmartSupplyChainDataGenerator: 
        # Define customer segments

        self.segment_configs = {

            'luxury_buyers': {

                'size': 300,

                'size': 30000,

                'purchase_frequency_mean': 18.0,  # Buy every 18 months

                'purchase_frequency_std': 6.0,

                'price_sensitivity_alpha': 2,
@@ -173,7 +173,7 @@ class SmartSupplyChainDataGenerator: 
                'lifetime_value': 2000

            },

            'sport_enthusiasts': {

                'size': 500,

                'size': 50000,

                'purchase_frequency_mean': 14.0,  # Buy every 14 months

                'purchase_frequency_std': 5.0,

                'price_sensitivity_alpha': 4,
@@ -187,7 +187,7 @@ class SmartSupplyChainDataGenerator: 
                'lifetime_value': 800

            },

            'casual_shoppers': {

                'size': 800,

                'size': 80000,

                'purchase_frequency_mean': 10.0,  # Buy every 10 months

                'purchase_frequency_std': 4.0,

                'price_sensitivity_alpha': 6,
@@ -462,15 +462,15 @@ def main(): 
    full_dataset = generator.generate_dataset(years=11)



    # Save full dataset

    generator.save_dataset(full_dataset, 'data/sim_supply_chain_data_full.json')

    generator.save_dataset(full_dataset, 'data/sim2_supply_chain_data_full.json')



    # Save training data (10 years)

    training_data = generator.get_training_data(full_dataset, training_years=10)

    generator.save_dataset(training_data, 'data/sim_supply_chain_data_training.json')

    generator.save_dataset(training_data, 'data/sim2_supply_chain_data_training.json')



    # Save test data (year 11)

    test_data = generator.get_test_data(full_dataset, test_year=11)

    with open('data/sim_supply_chain_data_test.json', 'w') as f:

    with open('data/sim2_supply_chain_data_test.json', 'w') as f:

        json.dump(test_data, f, indent=2)



    print("\n" + "=" * 60)