Skip to content

Feature Configuration

Feature Configuration

The feature configuration allows you to chose what feature to use, and to adjust their parameters to your needs. Follow the steps below to create a YAML configuration file

1. YAML Configuration Structure

The configuration file should have the following structure

  - name: Section Name
      - FeatureName1
      - FeatureName2
      - FeatureName3
  • report_sections: A list of sections that will appear in the final report. Each section consists of a name and a list of features to be included.
  • name: The name of the section.
  • features: The list of feature names to be included in the section.

2. Available Features

Please refer to the default configuration files to explore the available features and their names.

For a more in-depth explanation of each feature, please check out this page.

3. Feature Customization

Each feature can be customized by providing additional arguments in the configuration. Here's an example of customizing the DetectionSampleVisualization feature:

  - name: Object Detection Features
      - DetectionSampleVisualization:
          n_rows: 6
          n_cols: 2
          stack_splits_vertically: true
      - ... # Add any other features here
  • DetectionSampleVisualization: The feature name.
  • n_rows, n_cols: The number of rows and columns to use for displaying samples.
  • stack_splits_vertically: Whether to show train/test samples vertically or side by side.

4. Using the Configuration

To use the configuration, provide the path of your YAML file to the relevant analysis manager. For example, for object detection analysis:

from data_gradients.managers.detection_manager import DetectionAnalysisManager

train_data = ...
val_data = ...
class_names = ...

analyzer = DetectionAnalysisManager(
    report_title="Testing Data-Gradients Object Detection",
    config_path="path/to/custom-detection-config.yaml"  # Add this parameter to the manager initialization

By following these steps, you can easily customize the features included in your analysis report and fine-tune their parameters according to your requirements.