Update data_generation.py
Browse files- 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("
|
| 26 |
-
poor_generator = CTGANSynthesizer.load("
|
| 27 |
-
standard_generator = CTGANSynthesizer.load("
|
| 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)
|