luisejdm commited on
Commit
efada2f
·
verified ·
1 Parent(s): dcab61f

Update data_generation.py

Browse files
Files changed (1) hide show
  1. data_generation.py +3 -3
data_generation.py CHANGED
@@ -10,9 +10,9 @@ def generate_synthetic_training_data(n=30_000):
10
  Returns:
11
  pd.DataFrame: The generated synthetic training data.
12
  """
13
- good_generator = CTGANSynthesizer.load("../models/v4/synth_good.pkl")
14
- poor_generator = CTGANSynthesizer.load("../models/v4/synth_poor.pkl")
15
- standard_generator = CTGANSynthesizer.load("../models/v4/synth_standard.pkl")
16
 
17
  synth_good = good_generator.sample(n)
18
  synth_poor = poor_generator.sample(n)
 
10
  Returns:
11
  pd.DataFrame: The generated synthetic training data.
12
  """
13
+ good_generator = CTGANSynthesizer.load("models/v4/synth_good.pkl")
14
+ poor_generator = CTGANSynthesizer.load("models/v4/synth_poor.pkl")
15
+ standard_generator = CTGANSynthesizer.load("models/v4/synth_standard.pkl")
16
 
17
  synth_good = good_generator.sample(n)
18
  synth_poor = poor_generator.sample(n)