OpenFE项目架构解析与快速入门指南

【免费下载链接】OpenFEOpenFE: automated feature generation with expert-level performance项目地址: https://gitcode.com/gh_mirrors/op/OpenFE

OpenFE是一个高效的自动化特征生成工具，专为表格数据设计。该项目采用模块化架构，通过智能算法自动发现并生成能够提升机器学习模型性能的新特征。

项目架构层次解析

OpenFE项目采用清晰的分层架构设计，便于理解和使用：

核心引擎层

openfe/目录包含项目的核心功能模块，主要包括：

openfe.py：主要的OpenFE类实现，负责特征生成的核心流程
FeatureGenerator.py：特征生成器，实现各种特征操作符
FeatureSelector.py：特征选择器，优化生成的特征集合
utils.py：工具函数，提供辅助功能支持

文档资源层

docs/目录提供完整的项目文档体系，包含详细的API说明和最佳实践指南。文档采用Sphinx构建，支持多格式输出。

代码示例层

examples/目录包含丰富的使用案例，从基础的加利福尼亚房价预测到复杂的IEEE-CIS欺诈检测竞赛，覆盖从入门到实战的各个应用场景。

核心启动机制

OpenFE的启动核心是setup.py文件，它承担着项目依赖管理和分发部署的重要职责：

from setuptools import setup, find_packages setup( name="openfe", version="0.0.8", author="Tianping Zhang", description="OpenFE: automated feature generation beyond expert-level performance", install_requires=[ "numpy>=1.19.3", "pandas>=1.1.5", "scikit_learn>=0.24.2", "lightgbm>=3.3.2", "scipy>=1.5.4", "tqdm", "pyarrow", ], )

快速入门实战

环境安装

使用pip进行快速安装：

pip install openfe

基础使用示例

只需四行代码即可完成特征生成：

from openfe import OpenFE, transform ofe = OpenFE() features = ofe.fit(data=train_x, label=train_y, n_jobs=n_jobs) train_x, test_x = transform(train_x, test_x, features, n_jobs=n_jobs)

完整工作流程

以下是一个完整的加利福尼亚房价预测示例：

import pandas as pd from sklearn.datasets import fetch_california_housing from openfe import OpenFE, transform from sklearn.model_selection import train_test_split import lightgbm as lgb from sklearn.metrics import mean_squared_error # 数据准备 data = fetch_california_housing(as_frame=True).frame label = data[['MedHouseVal']] del data['MedHouseVal'] train_x, test_x, train_y, test_y = train_test_split(data, label, test_size=0.2, random_state=1) # 基准性能评估 score = get_score(train_x, test_x, train_y, test_y) print("特征生成前的MSE：", score) # 特征生成 ofe = OpenFE() ofe.fit(data=train_x, label=train_y, n_jobs=4) # 应用生成的特征 train_x, test_x = transform(train_x, test_x, ofe.new_features_list[:10], n_jobs=4) score = get_score(train_x, test_x, train_y, test_y) print("特征生成后的MSE：", score) # 输出生成的特征 print("前10个生成的特征：") for feature in ofe.new_features_list[:10]: print(tree_to_formula(feature))

核心技术特性

OpenFE具备多项技术优势：

广泛的适用性

支持二分类、多分类和回归任务
兼容GBDT和神经网络模型
自动处理缺失值和分类特征

高效的算法设计

包含23个有效特征操作符
支持并行计算
采用连续特征减半策略优化计算效率

专家级性能在IEEE-CIS欺诈检测Kaggle竞赛中，使用OpenFE生成特征的简单XGBoost模型击败了6351个数据科学团队中的99.3%

高级功能配置

特征提升功能

通过feature_boosting参数启用特征提升，进一步优化特征质量：

ofe = OpenFE() features = ofe.fit(data=train_x, label=train_y, n_jobs=4, feature_boosting=True)

自定义候选特征

用户可以根据先验知识自定义候选特征列表：

candidate_features = ofe.get_candidate_features( numerical_features=['feature1', 'feature2'], categorical_features=['cat_feature1'], ordinal_features=['ordinal_feature1'], order=1 )