luisejdm commited on
Commit
91ce837
·
verified ·
1 Parent(s): 9301d5d

Update data_generation.py

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