Registry / ai-ml / autogluon-features

autogluon-features

JSON →
library1.6.1pypypi✓ verified 24d ago

The `autogluon-features` sub-package provides core machine learning feature engineering capabilities for the AutoGluon AutoML library. It handles tasks like detecting data types, transforming categorical, datetime, and text features, and managing feature metadata. This package is usually consumed internally by AutoGluon's higher-level predictors, but can be used directly for advanced customization. It is currently at version 1.5.0 and releases in conjunction with the main AutoGluon library.

pip install autogluon-features
INSTALL
IMPORT
SIG · AUTOGLUON-FEATURES
A
autogluon-features
ai-mlpythonv1.6.1
Install
Import
Disk
Pass rate
0/ 10
Env Coverage0 / 10
glibc
3.93.13
musl
3.93.13
Install & Compatibility
Where this runs
tested against v1.6.1 · pip install
no network on importno background threads
Install × environment matrix
Each cell = how many times install + import succeeded across repeated harness runs. Partial = flaky.
glibc = Debian/Ubuntu slim · musl = Alpine Linux
musl
glibc
py 3.10
✕ build_error
1/2 runs
py 3.11
✕ build_error
1/2 runs
py 3.12
✕ build_error
1/2 runs
py 3.13
✕ build_error
1/2 runs
py 3.9
✕ build_error
1/2 runs
Code
Verified usage

Verified import paths — ran on the pinned version, not inferred.

PipelineFeatureGenerator
from autogluon.features import PipelineFeatureGenerator
from autogluon.features import PipelineFeatureGenerator

This quickstart demonstrates how to instantiate and use a `PipelineFeatureGenerator` from `autogluon-features` to apply common transformations like categorical encoding and datetime feature extraction to a pandas DataFrame.

import pandas as pd from autogluon.features.generators import PipelineFeatureGenerator, CategoryFeatureGenerator, DatetimeFeatureGenerator # Create a sample DataFrame data = { 'numeric_col': [1, 2, 3, 4, 5], 'categorical_col': ['A', 'B', 'A', 'C', 'B'], 'datetime_col': ['2023-01-01', '2023-01-02', '2023-01-03', '2023-01-04', '2023-01-05'], 'text_col': ['hello world', 'foo bar', 'hello again', 'world peace', 'bar foo'] } df = pd.DataFrame(data) # Initialize a pipeline of feature generators # CategoryFeatureGenerator converts object/category dtypes # DatetimeFeatureGenerator extracts year, month, day, etc. from datetime columns pipeline_generator = PipelineFeatureGenerator( [CategoryFeatureGenerator(), DatetimeFeatureGenerator(fillna_limit=0)] ) # Fit and transform the DataFrame df_transformed = pipeline_generator.fit_transform(X=df) print("Original DataFrame:") print(df) print("\nTransformed DataFrame features (first 5 rows):") print(df_transformed.head()) print("\nFeature metadata after transformation (output features):") print(pipeline_generator.feature_metadata_out.pretty_print())
Debug
Known issues
breakingLoading models trained with a different AutoGluon version (which includes `autogluon-features` components) is not supported and can lead to crashes, incorrect predictions, or unexpected behavior due to internal API changes and serialization formats.
fix
Always train and predict with the same AutoGluon (and therefore `autogluon-features`) version. When upgrading AutoGluon, retrain your models with the new version.
affects: All versions, explicitly highlighted in v0.8.2 onwards.
gotchaDirectly using `autogluon-features` components (like `FeatureGenerator` subclasses) is primarily for advanced customization. Most users should leverage feature engineering through AutoGluon's high-level APIs like `TabularPredictor`, which manage feature generation automatically.
fix
Unless you have a specific need to customize the feature generation pipeline, prefer using `AutoGluon.TabularPredictor.fit()` which handles feature processing implicitly.
affects: All versions.
breakingPython version compatibility has changed. Support for Python 3.8 was dropped in AutoGluon v1.2.0, and support for Python 3.12 was added. Current versions require Python >=3.10 and <3.14.
fix
Ensure your Python environment is within the supported range (currently 3.10-3.13) before installing or upgrading AutoGluon. Check `requires_python` on PyPI for the most up-to-date information.
affects: >=1.2.0
Upgrade
Version history
1.6.1latest on PyPI · released Aug 6, 2026
Audit
Dependencies

No dependency data recorded yet.

Agent activity
34 hits · last 30 days
node
30
OpenAI (training)
1
Resources