Spaces:
Sleeping
Sleeping
Update data_generation.py
Browse files- 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("
|
| 24 |
-
poor_generator = CTGANSynthesizer.load("
|
| 25 |
-
standard_generator = CTGANSynthesizer.load("
|
| 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)
|