Skip to main content

A comprehensive machine learning toolkit for data analysis, preprocessing, modeling, and evaluation.

Project description

FreeAeon-ML

FreeAeon-ML 是一个一站式的 Python 机器学习工具包,封装了常用的机器学习流程模块,包括数据探索分析、数据预处理、特征选择、模型训练(分类、回归、时间序列)、模型评估和可视化,旨在帮助研究者和工程师高效构建、训练和评估机器学习模型。


🚀 特性功能

  • 📊 数据探索与统计分析:正态性检验、分布拟合、相关性分析等
  • 🧹 数据预处理:标准化、异常值处理、Box-Cox 变换、分箱等
  • 🔍 特征选择:信息图谱、方差分析、PCA 降维、Granger 因果检验等
  • 🧠 模型训练支持
    • 分类模型:DT, RF, SVM, ANN, GLM, Naive Bayes, GBM, XGBoosting,...
    • 回归模型:RF, ANN, GLM, GBM, XGBoosting,...
    • 时间序列建模:自动 ARIMA 与分解分析
  • 📈 模型评估:评估指标自动输出、重要性排序
  • 💾 模型保存与加载
  • 🧬 样本均衡与增强:SMOTE 过采样、自动切分
  • 📊 可视化支持:热力图、等高线、桑基图、序列图等
  • ⚙️ H2O 引擎集成:高性能分布式模型支持

📦 安装方式

pip install FreeAeon-ML

✅ 环境依赖

  • Python >= 3.7
  • Java Runtime Environment (JRE) 8+
  • 主要依赖库:
    • numpy, pandas, matplotlib, seaborn
    • scipy, scikit-learn, statsmodels
    • h2o

📌 注意:必须安装 Java 环境! FreeAeon-ML 使用 H2O 平台进行部分模型训练,需确保系统已安装 Java:

java -version

若未安装,请参考以下方式:


🧪 快速示例

import numpy as np
import pandas as pd
from FreeAeonML.FADataPreprocess import CFADataPreprocess
from FreeAeonML.FASample import CFASample
from FreeAeonML.FAModelClassify import CFAModelClassify
from h2o.estimators import H2ORandomForestEstimator
import h2o

h2o.init()

# 生成样本数据
df_sample = CFASample.get_random_classification(1000, n_feature=5, n_class=2)
df_train, df_test = CFASample.split_dataset(df_sample)

# 定义模型

# 使用系统自带的模型进行训练
model = CFAModelClassify(models=None)

#如果需要使用指定的模型进行训练,请按照以下格式指定模型)
#model = CFAModelClassify(models={"rf": H2ORandomForestEstimator()})

model.train(df_train, y_column="y")

# 模型评估
df_pred = model.predict(df_test, y_column="y")
df_eval = model.evaluate(df_test, y_column="y")
print(df_eval)

📁 模块说明

模块名 描述
FADataEDA 探索性数据分析
FADataPreprocess 数据预处理(标准化、异常值等)
FAFeatureSelect 特征选择(PCA、因果性检验等)
FAModelClassify 分类模型训练封装
FAModelRegression 回归模型训练封装
FAModelSeries 时间序列建模(自动 ARIMA)
FAEvaluation 模型评估与指标输出
FAVisualize 可视化模块(热图、桑基图、等高线等)
FASample 样本生成与增强工具箱

🧪 测试脚本示例

测试脚本位于 tests/ 目录,支持以下演示:

  • demo_DataEDA.py:数据分析演示
  • demo_DataPreprocess.py:预处理功能测试
  • demo_FeatureSelect.py:特征选择测试
  • demo_ModelClassify.py:分类模型演示
  • demo_ModelRegression.py:回归模型演示
  • demo_ModelSeries.py:时间序列建模演示
  • demo_Sample.py:样本生成与增强测试
  • demo_Visualize.py:图形可视化测试

运行示例:

python tests/demo_ModelClassify.py

📄 License

FreeAeon-ML is released under the MIT License.
© 2025 FreeAeon Contributors


🤝 欢迎贡献

欢迎 PR、Issue 与建议!请确保代码规范、清晰,附带测试。


✍️ Author

Jim Xie
📧 E-Mail: jim.xie.cn@outlook.com, xiewenwei@sina.com
🔗 GitHub: https://github.com/jim-xie-cn/FreeAeon-ML


🧠 Citation

If you use this project in academic work, please cite it as:

Jim Xie, FreeAeon-ML: A comprehensive machine learning toolkit for data analysis, preprocessing, modeling, and evaluation., 2025.
GitHub Repository: https://github.com/jim-xie-cn/FreeAeon-ML

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

FreeAeon-ML-0.1.2.tar.gz (23.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

FreeAeon_ML-0.1.2-py3-none-any.whl (29.4 kB view details)

Uploaded Python 3

File details

Details for the file FreeAeon-ML-0.1.2.tar.gz.

File metadata

  • Download URL: FreeAeon-ML-0.1.2.tar.gz
  • Upload date:
  • Size: 23.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.8.0 pkginfo/1.10.0 readme-renderer/34.0 requests/2.27.1 requests-toolbelt/1.0.0 urllib3/1.26.20 tqdm/4.64.1 importlib-metadata/4.8.3 keyring/23.4.1 rfc3986/1.5.0 colorama/0.4.5 CPython/3.6.15

File hashes

Hashes for FreeAeon-ML-0.1.2.tar.gz
Algorithm Hash digest
SHA256 a1c827838d5c06e7e5d9f0608990fcbfdc4f6d860b53d96043cf0c7cf6677cf9
MD5 75254c5364c2c3b1589ba67397a52fba
BLAKE2b-256 af7e912b22acadb9665def2c4a2064787213cf4beba5ae00cae6c73c14f543af

See more details on using hashes here.

File details

Details for the file FreeAeon_ML-0.1.2-py3-none-any.whl.

File metadata

  • Download URL: FreeAeon_ML-0.1.2-py3-none-any.whl
  • Upload date:
  • Size: 29.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.8.0 pkginfo/1.10.0 readme-renderer/34.0 requests/2.27.1 requests-toolbelt/1.0.0 urllib3/1.26.20 tqdm/4.64.1 importlib-metadata/4.8.3 keyring/23.4.1 rfc3986/1.5.0 colorama/0.4.5 CPython/3.6.15

File hashes

Hashes for FreeAeon_ML-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 e70999bfabcac6d7a6aafb4f478d445f380587db5f7243acb3f872a0b2e5447b
MD5 78f91dafae81ddc65b87ff005cbba8a0
BLAKE2b-256 34a6d5bce4d845105ab5636f73acd02d8d358687f84fc475017056fde897b6fe

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page