gretel-synthetics-0.10.0-part3

train_and_gen.py

# ``batcher`` is our ``DataFrameBatch`` instance

# this creates a DataFrame and a "train.csv" file for each of our batches
batcher.create_training_data()

# loops over all batches, and generates the models
batcher.train_all_batches()

# loops over all models, and generates synthetic lines for each one
#
# returns a dictionary that shows if each batch was able to generate
# the full number of requested lines
#
# Example return:
# {
#   0: True,
#   1: True,
#   2: True,
#   ...
# }
status = batcher.generate_all_batch_lines()

# If all batches could generate the requested number of lines...Shizzam!
synthetic_df = batcher.batches_to_df()