File size: 483 Bytes
685d968 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import json
from synthetic_data.pipeline import SyntheticDataPipeline
def generate_sample():
pipeline = SyntheticDataPipeline()
print("Generating sample batch...")
results = pipeline.run_batch(count=2, category="company.brand_core")
with open("synthetic_data/sample_batch.json", "w") as f:
json.dump(results, f, indent=2)
print(f"Saved {len(results)} examples to synthetic_data/sample_batch.json")
if __name__ == "__main__":
generate_sample()
|