---
title: "Augment Dataset"
method: POST
path: "/felix/dataset/augment"
tags: ["felix"]
---

# Augment Dataset

`POST /felix/dataset/augment`

Augment a dataset by removing duplicates/outliers and generating synthetic samples.

This endpoint creates a NEW dataset with the augmentations applied.
Operations include:
- **remove_duplicates**: Remove exact duplicate samples.
- **remove_outliers**: Remove samples with anomalous lengths.
- **balance**: Generate synthetic samples for underrepresented classes/entities.

The original dataset is preserved.

## Request body

- DatasetAugmentationRequest — Request for dataset augmentation.
  - `task_type` 'ner' | 'classification' | 'custom', required — Task type of the dataset
  - `dataset_name` string, nullable — Name of stored dataset to augment
  - `dataset_version` string, nullable — Dataset version (latest if omitted)
  - `dataset` object[], nullable — Inline dataset to augment (if dataset_name not provided)
  - `operations` AugmentationOperation[], required — List of augmentation operations to perform
    - `type` 'remove_duplicates' | 'remove_outliers' | 'balance', required — Type of augmentation operation
    - `enabled` boolean — Whether this operation is enabled
  - `target_distribution` object, nullable — Target class/entity distribution as percentages (must sum to 1.0)
  - `domain_description` string, nullable — Domain description for synthetic sample generation
  - `new_dataset_name` string, nullable — Name for the new augmented dataset (required if update_in_place=False)
  - `labels` string[], nullable — Labels for generation (required if balance operation used)
  - `update_in_place` boolean — If True, creates new version with same name and soft-deletes old version

## Response `200`

Successful Response

- DatasetAugmentationResponse — Response containing augmentation results.
  - `success` boolean, required
  - `original_dataset_name` string, nullable
  - `original_dataset_version` string, nullable
  - `new_dataset` DatasetResponse, required — Response model for a single dataset.
    - `id` string, required
    - `user_id` string, required
    - `dataset_name` string, required
    - `dataset_path` string, required
    - `dataset_type` string, required
    - `size` integer, nullable
    - `sample_size` integer, nullable
    - `train_ratio` number, nullable — Train split ratio for this dataset version. Left-to-right split with no shuffle; validation is the tail.
    - `created_at` string, required
    - `updated_at` string, required
    - `version_number` string
    - `root_dataset_id` string, nullable
    - `project_id` string, nullable
    - `schema` object, nullable
    - `schema_warnings` string[], nullable
    - `validation` object, nullable
    - `annotation_status` 'none' | 'in_progress' | 'completed', nullable
    - `annotation_config` object, nullable
    - `annotation_progress` object, nullable
    - `status` 'initialized' | 'uploading' | 'converting' | 'validating' | 'ready' | 'failed' | 'generating' | 'queued', nullable — Dataset status: initialized/uploading/converting/validating/ready/failed/generating/queued
    - `processing_error` string, nullable — Error message if status is failed
    - `type` string — Dataset purpose tag: 'training', 'evaluation', or 'benchmark'
    - `visibility` string — Dataset visibility: 'private' or 'public'
    - `is_competition` boolean — Whether this dataset is a competition benchmark
    - `labels` string[], nullable — Label names (entity types for NER, class labels for classification)
    - `generation_type` string, nullable — How the dataset was created: synthesize, upload, auto_relabel, manual_relabel, grow, external
    - `is_seed` boolean, nullable — Whether this dataset is a seed dataset (small set for review before full expansion)
    - `synthesis_session_id` string, nullable — UUID of the synthesis log session for this dataset, used to restore creation workflow on resume
    - `column_mapping` object, nullable — Column mapping from original to standard names
  - `modifications` ModificationSummary, required — Summary of modifications applied during augmentation.
    - `duplicates_removed` integer
    - `outliers_removed` integer
    - `samples_generated` integer
    - `original_count` integer
    - `final_count` integer
  - `distribution_comparison` DistributionComparison[], required
    - `label` string, required
    - `before_count` integer, required
    - `before_percentage` number, required
    - `after_count` integer, required
    - `after_percentage` number, required
  - `message` string, required
  - `updated_in_place` boolean — True if the original dataset was replaced (soft-deleted)

## Other responses

- `422` — Validation Error

---

[API](https://skmtc.net/pioneer/apis/brain-api.md) · [All operations](https://skmtc.net/pioneer/apis/brain-api/llms.txt) · [OpenAPI document](https://skmtc-service-staging.skmtc.workers.dev/v1/apis/pioneer/brain-api/revisions/31dfe831e079/schema)