luisejdm commited on
Commit
f624647
·
verified ·
1 Parent(s): e7ccdab

Update data_generation.py

Browse files
Files changed (1) hide show
  1. data_generation.py +3 -3
data_generation.py CHANGED
@@ -20,9 +20,9 @@ def generate_synthetic_training_data(n=30_000):
20
  Returns:
21
  pd.DataFrame: The generated synthetic training data.
22
  """
23
- good_generator = CTGANSynthesizer.load("../models/v4/synth_good.pkl")
24
- poor_generator = CTGANSynthesizer.load("../models/v4/synth_poor.pkl")
25
- standard_generator = CTGANSynthesizer.load("../models/v4/synth_standard.pkl")
26
 
27
  synth_good = good_generator.sample(n)
28
  synth_poor = poor_generator.sample(n)
 
20
  Returns:
21
  pd.DataFrame: The generated synthetic training data.
22
  """
23
+ good_generator = CTGANSynthesizer.load("models/v4/synth_good.pkl")
24
+ poor_generator = CTGANSynthesizer.load("models/v4/synth_poor.pkl")
25
+ standard_generator = CTGANSynthesizer.load("models/v4/synth_standard.pkl")
26
 
27
  synth_good = good_generator.sample(n)
28
  synth_poor = poor_generator.sample(n)