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