Skip to navigation

The Basics

View as Markdown

🎨 Data Designer Tutorial: The Basics

📚 What you'll learn

This notebook demonstrates the basics of Data Designer by generating a simple product review dataset.

📦 Import Data Designer

  • data_designer.config provides access to the configuration API.

  • DataDesigner is the main interface for data generation.

Python
1import data_designer.config as dd
2from data_designer.interface import DataDesigner
3

⚙️ Initialize the Data Designer interface

  • DataDesigner is the main object responsible for managing the data generation process.

  • When initialized without arguments, the default model providers are used.

Python
1data_designer = DataDesigner()
2

🎛️ Define model configurations

  • Each ModelConfig defines a model that can be used during the generation process.

  • The "model alias" is used to reference the model in the Data Designer config (as we will see below).

  • The "model provider" is the external service that hosts the model (see the model config docs for more details).

  • By default, we use build.nvidia.com as the model provider.

Python
1# This name is set in the model provider configuration.
2MODEL_PROVIDER = "nvidia"
3
4# The model ID is from build.nvidia.com.
5MODEL_ID = "nvidia/nemotron-3.5-lightning-30b-a3b"
6
7# We choose this alias to be descriptive for our use case.
8MODEL_ALIAS = "nemotron-lightning"
9
10model_configs = [
11 dd.ModelConfig(
12 alias=MODEL_ALIAS,
13 model=MODEL_ID,
14 provider=MODEL_PROVIDER,
15 inference_parameters=dd.ChatCompletionInferenceParams(
16 temperature=1.0,
17 top_p=0.95,
18 max_tokens=2048,
19 extra_body={"chat_template_kwargs": {"enable_thinking": False}},
20 ),
21 )
22]
23

🏗️ Initialize the Data Designer Config Builder

  • The Data Designer config defines the dataset schema and generation process.

  • The config builder provides an intuitive interface for building this configuration.

  • The list of model configs is provided to the builder at initialization.

Python
1config_builder = dd.DataDesignerConfigBuilder(model_configs=model_configs)
2

🎲 Getting started with sampler columns

  • Sampler columns offer non-LLM based generation of synthetic data.

  • They are particularly useful for steering the diversity of the generated data, as we demonstrate below.


You can view available samplers using the config builder's info property:

Python
1config_builder.info.display("samplers")
2
Output
─────────────────────────────────────────── NeMo Data Designer Samplers ───────────────────────────────────────────

┏━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━┓
┃ Type               ┃ Parameter                ┃ Data Type                         ┃ Required ┃ Constraints      ┃
┡━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━┩
│ bernoulli          │ p                        │ number                            │    ✓     │ >= 0.0, <= 1.0   │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ bernoulli_mixture  │ p                        │ number                            │    ✓     │ >= 0.0, <= 1.0   │
│                    │ dist_name                │ string                            │    ✓     │                  │
│                    │ dist_params              │ dict                              │    ✓     │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ binomial           │ n                        │ integer                           │    ✓     │                  │
│                    │ p                        │ number                            │    ✓     │ >= 0.0, <= 1.0   │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ category           │ values                   │ string[] | integer[] | number[]   │    ✓     │ len > 1          │
│                    │ weights                  │ number[] | null                   │          │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ datetime           │ start                    │ string                            │    ✓     │                  │
│                    │ end                      │ string                            │    ✓     │                  │
│                    │ unit                     │ string                            │          │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ gaussian           │ mean                     │ number                            │    ✓     │                  │
│                    │ stddev                   │ number                            │    ✓     │                  │
│                    │ decimal_places           │ integer | null                    │          │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ person             │ locale                   │ string                            │          │                  │
│                    │ sex                      │ string | null                     │          │                  │
│                    │ city                     │ string | string[] | null          │          │                  │
│                    │ age_range                │ integer[]                         │          │ len > 2, len < 2 │
│                    │ select_field_values      │ object | null                     │          │                  │
│                    │ with_synthetic_personas  │ boolean                           │          │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ person_from_faker  │ locale                   │ string                            │          │                  │
│                    │ sex                      │ string | null                     │          │                  │
│                    │ city                     │ string | string[] | null          │          │                  │
│                    │ age_range                │ integer[]                         │          │ len > 2, len < 2 │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ poisson            │ mean                     │ number                            │    ✓     │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ scipy              │ dist_name                │ string                            │    ✓     │                  │
│                    │ dist_params              │ dict                              │    ✓     │                  │
│                    │ decimal_places           │ integer | null                    │          │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ subcategory        │ category                 │ string                            │    ✓     │                  │
│                    │ values                   │ dict                              │    ✓     │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ timedelta          │ dt_min                   │ integer                           │    ✓     │ >= 0             │
│                    │ dt_max                   │ integer                           │    ✓     │ > 0              │
│                    │ reference_column_name    │ string                            │    ✓     │                  │
│                    │ unit                     │ string                            │          │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ uniform            │ low                      │ number                            │    ✓     │                  │
│                    │ high                     │ number                            │    ✓     │                  │
│                    │ decimal_places           │ integer | null                    │          │                  │
│                    │ sampler_type             │ string                            │          │                  │
├────────────────────┼──────────────────────────┼───────────────────────────────────┼──────────┼──────────────────┤
│ uuid               │ prefix                   │ string | null                     │          │                  │
│                    │ short_form               │ boolean                           │          │                  │
│                    │ uppercase                │ boolean                           │          │                  │
│                    │ sampler_type             │ string                            │          │                  │
└────────────────────┴──────────────────────────┴───────────────────────────────────┴──────────┴──────────────────┘

Let's start designing our product review dataset by adding product category and subcategory columns.

Python
1config_builder.add_column(
2 dd.SamplerColumnConfig(
3 name="product_category",
4 sampler_type=dd.SamplerType.CATEGORY,
5 params=dd.CategorySamplerParams(
6 values=[
7 "Electronics",
8 "Clothing",
9 "Home & Kitchen",
10 "Books",
11 "Home Office",
12 ],
13 ),
14 )
15)
16
17config_builder.add_column(
18 dd.SamplerColumnConfig(
19 name="product_subcategory",
20 sampler_type=dd.SamplerType.SUBCATEGORY,
21 params=dd.SubcategorySamplerParams(
22 category="product_category",
23 values={
24 "Electronics": [
25 "Smartphones",
26 "Laptops",
27 "Headphones",
28 "Cameras",
29 "Accessories",
30 ],
31 "Clothing": [
32 "Men's Clothing",
33 "Women's Clothing",
34 "Winter Coats",
35 "Activewear",
36 "Accessories",
37 ],
38 "Home & Kitchen": [
39 "Appliances",
40 "Cookware",
41 "Furniture",
42 "Decor",
43 "Organization",
44 ],
45 "Books": [
46 "Fiction",
47 "Non-Fiction",
48 "Self-Help",
49 "Textbooks",
50 "Classics",
51 ],
52 "Home Office": [
53 "Desks",
54 "Chairs",
55 "Storage",
56 "Office Supplies",
57 "Lighting",
58 ],
59 },
60 ),
61 )
62)
63
64config_builder.add_column(
65 dd.SamplerColumnConfig(
66 name="target_age_range",
67 sampler_type=dd.SamplerType.CATEGORY,
68 params=dd.CategorySamplerParams(values=["18-25", "25-35", "35-50", "50-65", "65+"]),
69 )
70)
71
72# Optionally validate that the columns are configured correctly.
73data_designer.validate(config_builder)
74
Output
[19:55:07] [INFO] ✅ Validation passed

Next, let's add samplers to generate data related to the customer and their review.

Python
1config_builder.add_column(
2 dd.SamplerColumnConfig(
3 name="customer",
4 sampler_type=dd.SamplerType.PERSON_FROM_FAKER,
5 params=dd.PersonFromFakerSamplerParams(age_range=[18, 70], locale="en_US"),
6 )
7)
8
9config_builder.add_column(
10 dd.SamplerColumnConfig(
11 name="number_of_stars",
12 sampler_type=dd.SamplerType.UNIFORM,
13 params=dd.UniformSamplerParams(low=1, high=5),
14 convert_to="int", # Convert the sampled float to an integer.
15 )
16)
17
18config_builder.add_column(
19 dd.SamplerColumnConfig(
20 name="review_style",
21 sampler_type=dd.SamplerType.CATEGORY,
22 params=dd.CategorySamplerParams(
23 values=["rambling", "brief", "detailed", "structured with bullet points"],
24 weights=[1, 2, 2, 1],
25 ),
26 )
27)
28
29data_designer.validate(config_builder)
30
Output
[19:55:07] [INFO] ✅ Validation passed

🦜 LLM-generated columns

  • The real power of Data Designer comes from leveraging LLMs to generate text, code, and structured data.

  • When prompting the LLM, we can use Jinja templating to reference other columns in the dataset.

  • As we see below, nested json fields can be accessed using dot notation.

Python
1config_builder.add_column(
2 dd.LLMTextColumnConfig(
3 name="product_name",
4 prompt=(
5 "You are a helpful assistant that generates product names. DO NOT add quotes around the product name.\n\n"
6 "Come up with a creative product name for a product in the '{{ product_category }}' category, focusing "
7 "on products related to '{{ product_subcategory }}'. The target age range of the ideal customer is "
8 "{{ target_age_range }} years old. Respond with only the product name, no other text."
9 ),
10 model_alias=MODEL_ALIAS,
11 )
12)
13
14config_builder.add_column(
15 dd.LLMTextColumnConfig(
16 name="customer_review",
17 prompt=(
18 "You are a customer named {{ customer.first_name }} from {{ customer.city }}, {{ customer.state }}. "
19 "You are {{ customer.age }} years old and recently purchased a product called {{ product_name }}. "
20 "Write a review of this product, which you gave a rating of {{ number_of_stars }} stars. "
21 "The style of the review should be '{{ review_style }}'. "
22 "Respond with only the review, no other text."
23 ),
24 model_alias=MODEL_ALIAS,
25 )
26)
27
28data_designer.validate(config_builder)
29
Output
[19:55:07] [INFO] ✅ Validation passed

🔁 Iteration is key – preview the dataset!

  1. Use the preview method to generate a sample of records quickly.

  2. Inspect the results for quality and format issues.

  3. Adjust column configurations, prompts, or parameters as needed.

  4. Re-run the preview until satisfied.

Python
1preview = data_designer.preview(config_builder, num_records=2)
2
Output
[19:55:07] [INFO] 🖼️ Preview generation in progress
[19:55:07] [INFO]   |-- 🔒 Jinja rendering engine: secure
[19:55:07] [INFO] ✅ Validation passed
[19:55:07] [INFO] ⛓️ Sorting column configs into a Directed Acyclic Graph
[19:55:07] [INFO] Skipping model health checks because DATA_DESIGNER_SKIP_MODEL_HEALTH_CHECKS=1
[19:55:07] [INFO] ⚡ Using async task-queue preview
[19:55:07] [INFO] 📝 llm-text model config for column 'product_name'
[19:55:07] [INFO]   |-- model: 'nvidia/nemotron-3.5-lightning-30b-a3b'
[19:55:07] [INFO]   |-- model alias: 'nemotron-lightning'
[19:55:07] [INFO]   |-- model provider: 'nvidia'
[19:55:07] [INFO]   |-- inference parameters:
[19:55:07] [INFO]   |  |-- generation_type=chat-completion
[19:55:07] [INFO]   |  |-- max_parallel_requests=4
[19:55:07] [INFO]   |  |-- extra_body={'chat_template_kwargs': {'enable_thinking': False}}
[19:55:07] [INFO]   |  |-- temperature=1.00
[19:55:07] [INFO]   |  |-- top_p=0.95
[19:55:07] [INFO]   |  |-- max_tokens=2048
[19:55:07] [INFO] 📝 llm-text model config for column 'customer_review'
[19:55:07] [INFO]   |-- model: 'nvidia/nemotron-3.5-lightning-30b-a3b'
[19:55:07] [INFO]   |-- model alias: 'nemotron-lightning'
[19:55:07] [INFO]   |-- model provider: 'nvidia'
[19:55:07] [INFO]   |-- inference parameters:
[19:55:07] [INFO]   |  |-- generation_type=chat-completion
[19:55:07] [INFO]   |  |-- max_parallel_requests=4
[19:55:07] [INFO]   |  |-- extra_body={'chat_template_kwargs': {'enable_thinking': False}}
[19:55:07] [INFO]   |  |-- temperature=1.00
[19:55:07] [INFO]   |  |-- top_p=0.95
[19:55:07] [INFO]   |  |-- max_tokens=2048
[19:55:07] [INFO] ⚡️ Async generation: 2 column(s) (column 'product_name', column 'customer_review'), 4 tasks across 1 row group(s)
[19:55:07] [INFO] 🚀 (1/1) Dispatching with 2 records
[19:55:07] [INFO] 🎲 (1/1) Preparing samplers to generate 2 records across 6 columns
[19:55:11] [INFO] 📊 Progress [4.2s]:
[19:55:11] [INFO]   |-- ☀️ column 'product_name': 2/2 (100%) 0.5 rec/s
[19:55:11] [INFO]   |-- 🐔 column 'customer_review': 2/2 (100%) 0.5 rec/s
[19:55:11] [INFO] ✅ Async generation complete [4.2s]: 4 ok, 0 failed across 2 column(s)
[19:55:11] [INFO] 📊 Model usage summary:
[19:55:11] [INFO]   |-- model: nvidia/nemotron-3.5-lightning-30b-a3b
[19:55:11] [INFO]   |-- tokens: input=356, output=139, reasoning=0, total=495, tps=117
[19:55:11] [INFO]   |-- requests: success=4, failed=0, total=4, rpm=57
[19:55:11] [INFO] 📐 Measuring dataset column statistics:
[19:55:11] [INFO]   |-- 🎲 column: 'product_category'
[19:55:11] [INFO]   |-- 🎲 column: 'product_subcategory'
[19:55:11] [INFO]   |-- 🎲 column: 'target_age_range'
[19:55:11] [INFO]   |-- 🎲 column: 'customer'
[19:55:11] [INFO]   |-- 🎲 column: 'number_of_stars'
[19:55:11] [INFO]   |-- 🎲 column: 'review_style'
[19:55:11] [INFO]   |-- 📝 column: 'product_name'
[19:55:11] [INFO]   |-- 📝 column: 'customer_review'
[19:55:11] [INFO] 🙌 Preview complete!
Python
1# Run this cell multiple times to cycle through the 2 preview records.
2preview.display_sample_record()
3
Output
[index: 0]
                                                                                                              
                                              Generated Columns                                               
┏━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
┃ Name                    ┃ Value                                                                            ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┩
│ product_category        │ Books                                                                            │
├─────────────────────────┼──────────────────────────────────────────────────────────────────────────────────┤
│ product_subcategory     │ Non-Fiction                                                                      │
├─────────────────────────┼──────────────────────────────────────────────────────────────────────────────────┤
│ target_age_range        │ 65+                                                                              │
├─────────────────────────┼──────────────────────────────────────────────────────────────────────────────────┤
│ customer                │ {                                                                                │
│                         │     'uuid': '98898411-29f9-4650-90e6-7a272fba1045',                              │
│                         │     'locale': 'en_US',                                                           │
│                         │     'first_name': 'Robert',                                                      │
│                         │     'last_name': 'Decker',                                                       │
│                         │     'middle_name': None,                                                         │
│                         │     'sex': 'Male',                                                               │
│                         │     'street_number': '0378',                                                     │
│                         │     'street_name': 'Robert Summit',                                              │
│                         │     'city': 'Lake Jamesmouth',                                                   │
│                         │     'state': 'South Carolina',                                                   │
│                         │     'postcode': '02831',                                                         │
│                         │     'age': 49,                                                                   │
│                         │     'birth_date': '1977-02-14',                                                  │
│                         │     'country': 'South Africa',                                                   │
│                         │     'marital_status': 'married_present',                                         │
│                         │     'education_level': 'doctorate',                                              │
│                         │     'unit': '',                                                                  │
│                         │     'occupation': 'Corporate investment banker',                                 │
│                         │     'phone_number': '952.991.8873x881',                                          │
│                         │     'bachelors_field': 'business'                                                │
│                         │ }                                                                                │
├─────────────────────────┼──────────────────────────────────────────────────────────────────────────────────┤
│ number_of_stars         │ 2                                                                                │
├─────────────────────────┼──────────────────────────────────────────────────────────────────────────────────┤
│ review_style            │ structured with bullet points                                                    │
├─────────────────────────┼──────────────────────────────────────────────────────────────────────────────────┤
│ product_name            │ Golden Chapters: A Life In Stories                                               │
├─────────────────────────┼──────────────────────────────────────────────────────────────────────────────────┤
│ customer_review         │ • The concept of a life told in stories sounded promising and unique.            │
│                         │ • Unfortunately, the narrative flow felt fragmented and disjointed.              │
│                         │ • I struggled to connect with the characters or the overarching plot.            │
│                         │ • The execution felt rushed, leaving many questions unanswered.                  │
│                         │ • For the price, I expected a more polished and cohesive experience.             │
│                         │ • It left me feeling underwhelmed and longing for more depth.                    │
└─────────────────────────┴──────────────────────────────────────────────────────────────────────────────────┘
                                                                                                              
Python
1# The preview dataset is available as a pandas DataFrame.
2preview.dataset
3
Output
product_category product_subcategory target_age_range customer number_of_stars review_style product_name customer_review
0 Books Non-Fiction 65+ {'uuid': '98898411-29f9-4650-90e6-7a272fba1045... 2 structured with bullet points Golden Chapters: A Life In Stories • The concept of a life told in stories sounde...
1 Electronics Smartphones 18-25 {'uuid': '568f1dc4-2d77-46c9-86b7-eabf2563b1c1... 2 brief EchoBand This band is okay, but not worth the price. Fo...

📊 Analyze the generated data

  • Data Designer automatically generates a basic statistical analysis of the generated data.

  • This analysis is available via the analysis property of generation result objects.

Python
1# Print the analysis as a table.
2preview.analysis.to_report()
3
Output
──────────────────────────────────────── 🎨 Data Designer Dataset Profile ─────────────────────────────────────────

                                                                                                                   
                                                 Dataset Overview                                                  
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
┃ number of records               ┃ number of columns               ┃ percent complete records                    ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┩
│ 2                               │ 8                               │ 100.0%                                      │
└─────────────────────────────────┴─────────────────────────────────┴─────────────────────────────────────────────┘
                                                                                                                   
                                                                                                                   
                                                🎲 Sampler Columns                                                 
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
┃ column name                    ┃       data type ┃            number unique values ┃               sampler type ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━┩
│ product_category               │          string │                      2 (100.0%) │                   category │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ product_subcategory            │          string │                      2 (100.0%) │                subcategory │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ target_age_range               │          string │                      2 (100.0%) │                   category │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ customer                       │            dict │                      2 (100.0%) │          person_from_faker │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ number_of_stars                │             int │                       1 (50.0%) │                    uniform │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ review_style                   │          string │                      2 (100.0%) │                   category │
└────────────────────────────────┴─────────────────┴─────────────────────────────────┴────────────────────────────┘
                                                                                                                   
                                                                                                                   
                                                📝 LLM-Text Columns                                                
┏━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━┓
┃                       ┃               ┃                            ┃     prompt tokens ┃      completion tokens ┃
┃ column name           ┃     data type ┃       number unique values ┃        per record ┃             per record ┃
┡━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━┩
│ product_name          │        string │                 2 (100.0%) │      73.5 +/- 0.5 │            4.5 +/- 3.5 │
├───────────────────────┼───────────────┼────────────────────────────┼───────────────────┼────────────────────────┤
│ customer_review       │        string │                 2 (100.0%) │      71.0 +/- 4.0 │          60.0 +/- 25.5 │
└───────────────────────┴───────────────┴────────────────────────────┴───────────────────┴────────────────────────┘
                                                                                                                   
                                                                                                                   
╭────────────────────────────────────────────────── Table Notes ──────────────────────────────────────────────────╮
│                                                                                                                 │
│  1. All token statistics are based on a sample of max(1000, len(dataset)) records.                              │
│  2. Tokens are calculated using tiktoken's cl100k_base tokenizer.                                               │
│                                                                                                                 │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
                                                                                                                   
───────────────────────────────────────────────────────────────────────────────────────────────────────────────────

🆙 Scale up!

  • Happy with your preview data?

  • Use the create method to submit larger Data Designer generation jobs.

Python
1results = data_designer.create(config_builder, num_records=10, dataset_name="tutorial-1")
2
Output
[19:55:11] [INFO] OpenTelemetry metrics available at http://127.0.0.1:9464/metrics
[19:55:11] [INFO] 🎨 Creating Data Designer dataset
[19:55:11] [INFO]   |-- 🔒 Jinja rendering engine: secure
[19:55:11] [INFO] ✅ Validation passed
[19:55:11] [INFO] ⛓️ Sorting column configs into a Directed Acyclic Graph
[19:55:11] [INFO] Skipping model health checks because DATA_DESIGNER_SKIP_MODEL_HEALTH_CHECKS=1
[19:55:11] [INFO] ⚡ Using async task-queue builder
[19:55:11] [INFO] 📝 llm-text model config for column 'product_name'
[19:55:11] [INFO]   |-- model: 'nvidia/nemotron-3.5-lightning-30b-a3b'
[19:55:11] [INFO]   |-- model alias: 'nemotron-lightning'
[19:55:11] [INFO]   |-- model provider: 'nvidia'
[19:55:11] [INFO]   |-- inference parameters:
[19:55:11] [INFO]   |  |-- generation_type=chat-completion
[19:55:11] [INFO]   |  |-- max_parallel_requests=4
[19:55:11] [INFO]   |  |-- extra_body={'chat_template_kwargs': {'enable_thinking': False}}
[19:55:11] [INFO]   |  |-- temperature=1.00
[19:55:11] [INFO]   |  |-- top_p=0.95
[19:55:11] [INFO]   |  |-- max_tokens=2048
[19:55:11] [INFO] 📝 llm-text model config for column 'customer_review'
[19:55:11] [INFO]   |-- model: 'nvidia/nemotron-3.5-lightning-30b-a3b'
[19:55:11] [INFO]   |-- model alias: 'nemotron-lightning'
[19:55:11] [INFO]   |-- model provider: 'nvidia'
[19:55:11] [INFO]   |-- inference parameters:
[19:55:11] [INFO]   |  |-- generation_type=chat-completion
[19:55:11] [INFO]   |  |-- max_parallel_requests=4
[19:55:11] [INFO]   |  |-- extra_body={'chat_template_kwargs': {'enable_thinking': False}}
[19:55:11] [INFO]   |  |-- temperature=1.00
[19:55:11] [INFO]   |  |-- top_p=0.95
[19:55:11] [INFO]   |  |-- max_tokens=2048
[19:55:11] [INFO] ⚡️ Async generation: 2 column(s) (column 'product_name', column 'customer_review'), 20 tasks across 1 row group(s)
[19:55:11] [INFO] 🚀 (1/1) Dispatching with 10 records
[19:55:11] [INFO] 🎲 (1/1) Preparing samplers to generate 10 records across 6 columns
[19:55:12] [WARNING] Observed retryable model-task error: kind=rate_limit, http_status=429; the row task will be deferred.
[19:55:17] [INFO] 📊 Progress [5.2s]:
[19:55:17] [INFO]   |-- 😊 column 'product_name': 9/10 (90%) 1.7 rec/s
[19:55:17] [INFO]   |-- 🌘 column 'customer_review': 3/10 (30%) 0.6 rec/s
[19:55:24] [INFO] 📊 Progress [12.1s]:
[19:55:24] [INFO]   |-- 😊 column 'product_name': 9/10 (90%) 0.7 rec/s
[19:55:24] [INFO]   |-- 🌗 column 'customer_review': 7/10 (70%) 0.6 rec/s
[19:55:30] [INFO] 📊 Progress [18.7s]:
[19:55:30] [INFO]   |-- 😊 column 'product_name': 9/10 (90%) 0.5 rec/s
[19:55:30] [INFO]   |-- 🌖 column 'customer_review': 9/10 (90%) 0.5 rec/s
[19:55:30] [INFO] 🔄 (1/1) Salvaging 1 deferred task(s)
[19:55:31] [INFO] 📊 Progress [19.8s]:
[19:55:31] [INFO]   |-- 🤩 column 'product_name': 10/10 (100%) 0.5 rec/s
[19:55:31] [INFO]   |-- 🌕 column 'customer_review': 10/10 (100%) 0.5 rec/s
[19:55:31] [INFO] ✅ Async generation complete [19.8s]: 20 ok, 0 failed across 2 column(s)
[19:55:31] [INFO] 📊 Model usage summary:
[19:55:31] [INFO]   |-- model: nvidia/nemotron-3.5-lightning-30b-a3b
[19:55:31] [INFO]   |-- tokens: input=1769, output=2020, reasoning=0, total=3789, tps=189
[19:55:31] [INFO]   |-- requests: success=20, failed=1, total=21, rpm=62
[19:55:32] [INFO] 📐 Measuring dataset column statistics:
[19:55:32] [INFO]   |-- 🎲 column: 'product_category'
[19:55:32] [INFO]   |-- 🎲 column: 'product_subcategory'
[19:55:32] [INFO]   |-- 🎲 column: 'target_age_range'
[19:55:32] [INFO]   |-- 🎲 column: 'customer'
[19:55:32] [INFO]   |-- 🎲 column: 'number_of_stars'
[19:55:32] [INFO]   |-- 🎲 column: 'review_style'
[19:55:32] [INFO]   |-- 📝 column: 'product_name'
[19:55:32] [INFO]   |-- 📝 column: 'customer_review'
Python
1# Load the generated dataset as a pandas DataFrame.
2dataset = results.load_dataset()
3
4dataset.head()
5
Output
product_category product_subcategory target_age_range customer number_of_stars review_style product_name customer_review
0 Clothing Activewear 65+ {'uuid': 'bc5098a2-a313-4280-a5a5-109777da9ea0... 4 structured with bullet points Silver-Stretch Mobility Tee *   **Price Point:** The price is very reasona...
1 Books Non-Fiction 65+ {'uuid': '5322b627-b5c7-4938-a520-368f89ec5954... 3 brief Golden Age Reflections: A Non-Fiction Anthology The anthology had its moments, but it felt a b...
2 Home & Kitchen Appliances 50-65 {'uuid': '5bae685f-a98a-4d66-8efe-d21045ff7fbc... 1 detailed Heritage Hearth Enthusiast I recently invested in the Heritage Hearth Ent...
3 Home Office Storage 35-50 {'uuid': '2c80c497-35d6-41ca-92c4-0518a9867d20... 1 brief Elara Locker Absolutely terrible. The Elara Locker arrived ...
4 Electronics Cameras 50-65 {'uuid': '41f7dab5-57b3-48d4-935a-d20105b03b84... 4 detailed VintageCapture Elite I recently had the pleasure of adding the Vint...
Python
1# Load the analysis results into memory.
2analysis = results.load_analysis()
3
4analysis.to_report()
5
Output
──────────────────────────────────────── 🎨 Data Designer Dataset Profile ─────────────────────────────────────────

                                                                                                                   
                                                 Dataset Overview                                                  
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
┃ number of records               ┃ number of columns               ┃ percent complete records                    ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┩
│ 10                              │ 8                               │ 100.0%                                      │
└─────────────────────────────────┴─────────────────────────────────┴─────────────────────────────────────────────┘
                                                                                                                   
                                                                                                                   
                                                🎲 Sampler Columns                                                 
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
┃ column name                    ┃       data type ┃            number unique values ┃               sampler type ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━┩
│ product_category               │          string │                       5 (50.0%) │                   category │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ product_subcategory            │          string │                     10 (100.0%) │                subcategory │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ target_age_range               │          string │                       5 (50.0%) │                   category │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ customer                       │            dict │                     10 (100.0%) │          person_from_faker │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ number_of_stars                │             int │                       4 (40.0%) │                    uniform │
├────────────────────────────────┼─────────────────┼─────────────────────────────────┼────────────────────────────┤
│ review_style                   │          string │                       3 (30.0%) │                   category │
└────────────────────────────────┴─────────────────┴─────────────────────────────────┴────────────────────────────┘
                                                                                                                   
                                                                                                                   
                                                📝 LLM-Text Columns                                                
┏━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━┓
┃                       ┃               ┃                            ┃     prompt tokens ┃      completion tokens ┃
┃ column name           ┃     data type ┃       number unique values ┃        per record ┃             per record ┃
┡━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━┩
│ product_name          │        string │                10 (100.0%) │      73.5 +/- 1.0 │            4.0 +/- 2.7 │
├───────────────────────┼───────────────┼────────────────────────────┼───────────────────┼────────────────────────┤
│ customer_review       │        string │                10 (100.0%) │      69.0 +/- 2.5 │        103.0 +/- 173.4 │
└───────────────────────┴───────────────┴────────────────────────────┴───────────────────┴────────────────────────┘
                                                                                                                   
                                                                                                                   
╭────────────────────────────────────────────────── Table Notes ──────────────────────────────────────────────────╮
│                                                                                                                 │
│  1. All token statistics are based on a sample of max(1000, len(dataset)) records.                              │
│  2. Tokens are calculated using tiktoken's cl100k_base tokenizer.                                               │
│                                                                                                                 │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
                                                                                                                   
───────────────────────────────────────────────────────────────────────────────────────────────────────────────────

⏭️ Next Steps

Now that you've seen the basics of Data Designer, check out the following notebooks to learn more about: