fuxion 0.0.4.2
Synthetic dataset generation and normalization functions powered by LLMs
License: MIT
Requires Python: <4.0,>=3.10

# fuxion

LangChain + LLM powered data generation and normalization functions.
fuxion helps you generate a fully synthetic dataset with LLM APIs to train a task-specific model you can run on your own GPU.
Preliminary models for name, price, and address standardization are available on [HuggingFace](https://huggingface.co/PragmaticMachineLearning).

![fuxion](/assets/fuxion.png)

# Table of Contents

- [Description](#description)
- [Installation](#installation)
- [Usage](#usage)
  - [Generation and Normalization](#generation-and-normalization)
  - [Template Structure](#template-structure)
  - [Pipelines](#pipelines)

# Description

fuxion is a Python package that provides you with a data generation and normalization pipeline which could be used for testing, normalization and training machine learning models. Using fuxion, you are able to generate sythetic data for different types of use cases -- all that's required is that you pass the right prompt to the chain and watch how things unfold :sunglasses:

# Installation

We recommend that you create a virtual environment before proceeding with the installation process as it would help to create an isolated environment for this project. After doing that, you can proceed with the installation by following the steps below.

- install via pip

  ```bash
  pip install fuxion
  ```

- Add the following to your bashrc file and replace "your-key" with your OpenAI API key:

  ```bash
  export OPENAI_API_KEY="your-key"
  ```

# Usage

The process of creating useful synthetic data involves two main steps: data generation and normalization. fuxion now combines data generation and normalization into a single streamlined process using structured output.

## Generation and Normalization

The generation process in fuxion uses a template file to guide the LLM in creating synthetic data. This template file contains instructions and placeholders for few-shot examples. Here's an overview of how generation works:

1. Create a template file with instructions for the type of data you want to generate.
2. Prepare a few-shot example file in JSON format.
3. Define the output structure for your data.
4. Use the `DatasetPipeline` class to generate and normalize data in one step.

The template file, few-shot examples, and output structure work together to produce high-quality, structured synthetic data.

This structure guides the LLM to produce data in the specified format, ensuring consistency and proper typing.

## Template Structure

For each generation task, a template file is required to guide the LLM on what to do. Below, we provide a brief overview of what the template files should look like.

##### Generator templates

```
Generate a list of U.S. postal addresses separated by double newlines.

Make them as realistic and diverse as possible.
Include some company address, P.O. boxes, apartment complexes, etc.
Ensure the addresses are fake.

{{few_shot}}

List:
```

- The first few lines tells the chain to generate addresses and contains a bunch of creative instructions that determines the quality of the results.

- `{{few_shot}}` tells the chain to get few-shot examples provided in the examples folder.

- `List` returns the results in a list

> The same convention should be followed when creating subsequent templates for various data generation tasks.

##### Normalizer templates

In the latest version of fuxion, normalization is integrated directly into the pipeline process using an `output_structure` parameter. This eliminates the need for separate normalization templates.

### Creating an Output Structure

The output structure is a key component in fuxion's generation process. It defines the format and types of the generated data, effectively combining generation and normalization. Here's an example of how to define an output structure:

```python
output_structure = {
    "field_name1": data_type,
    "field_name2": data_type,
    # ... more fields as needed
}
```

For example, for normalizing names:

```python
output_structure = {
    "title": str,
    "given": str,
    "middle": str,
    "surname": str,
    "suffix": str
}
```

Or for addresses:

```python
output_structure = {
    "house_number": int,
    "street": str,
    "city": str,
    "state": str,
    "zip_code": str
}
```

This structure guides the LLM in formatting the generated data, ensuring consistent and properly typed output. For details on how to use this in a pipeline, refer to the [Pipelines](#pipelines) section.

## Pipelines

The latest version of fuxion simplifies the normalization process by incorporating it directly into the pipeline using structured output. This removes the need for a separate normalization template, making it easier for users.

The `DatasetPipeline` class is the primary interface for generating synthetic datasets. It handles both generation and normalization in a single process. Here's an example of how to use it:

```python
from fuxion.pipelines import DatasetPipeline
from rich import print

output_structure = {
    "title": str,
    "given": str,
    "middle": str,
    "surname": str,
    "suffix": str
}

pipeline_chain = DatasetPipeline(
    generator_template="examples/name_generator/generator.template",
    few_shot_file="examples/name_generator/few_shot.json",
    output_structure=output_structure,
    dataset_name="name_pipeline",
    k=5,
    model_name="gpt-4o",
    cache=False,
    verbose=True,
    temperature=1.0,
)

result = pipeline_chain.execute()
print(result)
```

This pipeline generates a dataset of 5 names, formatted according to the specified output structure.
This replaces the previous `normalizer_template` parameter, simplifying the process and reducing complexity.
The generated data is automatically saved to a file named `name_dataset.json` in the `datasets` directory.

fuxion can be used to generate synthetic data for various use cases, including rapid product testing and machine learning model training.
By customizing the template, few-shot examples, and output structure, you can create diverse and realistic datasets tailored to your specific needs.

<b> Models supported </b>

- gpt-3.5-turbo
- gpt-4
- gpt-4o
- gpt-4o-mini

### Future work:

fuxion is still a work in progress, but it is a good starting point for anyone looking to generate synthetic data for testing and training machine learning models. We plan to add more features to fuxion in the future, including a seamless functionality for accurate data generation and normalization using various llms (locally hosted or via the huggingface api). For now, OpenAI's models are the most functional and reliable.

Feel free to contribute to this project by opening an issue or a pull request. We would love to hear your thoughts on how we can improve fuxion!