Skip to content

Feature/llm conditional textgen#90

Merged
Diyago merged 7 commits intomasterfrom
feature/llm-conditional-textgen
Mar 7, 2026
Merged

Feature/llm conditional textgen#90
Diyago merged 7 commits intomasterfrom
feature/llm-conditional-textgen

Conversation

@Diyago
Copy link
Copy Markdown
Owner

@Diyago Diyago commented Mar 7, 2026

No description provided.

google-labs-jules Bot and others added 7 commits June 24, 2025 20:49
Adds functionality to LLMGenerator to generate novel text values for specified columns, conditioned on the values of other attribute columns, while attempting to preserve attribute distributions.

Key changes:
- Modified `SamplerLLM` to accept `text_generating_columns` and `conditional_columns` parameters.
- Updated `generate_data` to handle conditional logic, using `be_great.impute` for non-text columns and direct LLM prompting via a new `_generate_via_prompt` helper for text columns.
- Implemented novelty checks for generated text.
- Added unit tests (mocked due to environment constraints for full runs) for the new functionality.
- Updated README.md with documentation for the new parameters and an example.

Note: Full testing was impeded by sandbox environment disk space limitations preventing complete dependency installation (primarily PyTorch and its CUDA components).
@Diyago Diyago merged commit 4579995 into master Mar 7, 2026
2 of 4 checks passed
@Diyago Diyago deleted the feature/llm-conditional-textgen branch March 7, 2026 12:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant