TPOT AutoML 回归

OS 包 > 表格数据 > TPOT AutoML 回归

此模型是通用的表格数据（仅限数值）回归模型，需要先进行训练才能用于预测。它依赖 TPOT 自动找到最佳模型。

TPOT 是一个 Python 自动化机器学习工具，可使用遗传编程来优化机器学习管道。TPOT 通过智能地探索成千上万个可能的管道来找到最适合您的数据的管道，从而自动化机器学习中最繁琐的部分。一旦 TPOT 完成搜索（或您厌倦了等待），它会为您提供找到的最佳管道的 Python 代码，以便您从中修改管道。TPOT 基于 scikit-learn 构建，因此它生成的所有代码对于 scikit-learn 用户来说应该都很熟悉。

模型详细信息

输入类型

JSON

输入说明

模型用于进行预测的特征。例如：{ “Feature1”: 12，“Feature2”: 222，...，“FeatureN”: 110}

输出说明

包含预测列表的 JSON：

示例：

{ "predictions" : "[12, 12, 2, 354, 12, 2] }

管道

此包支持所有三种类型的管道（完整训练、训练和评估）。

数据集格式

此 ML 包将在数据集（而不是子目录）中查找 csv 文件

csv 文件需要遵循以下两个规则：

• 数据的第一行必须包含标头/列名称。
• 所有列，必须为数值型（int、float）。模型无法执行特征编码，但能够执行目标编码。如果模型执行目标编码，则在预测时，模型还会返回目标变量的标签。

环境变量

• max_time_mins ：运行管道的时间（以分钟为单位）。 训练时间越长，TPOT 找到好的模型的机会就越大。 （默认值：2）
• target_column：目标列的名称（默认值：“target”）
• 打分：TPOT 使用 sklearn.model_selection.cross_val_score 来用于评估管道，从而为评分函数提供相同的支持（默认值：“accuracy”）。使用标准 scikit-learn 评分指标。
• keep_training：完成 TPOT 运行通常需要几个小时到几天（除非它是一个小型数据集），但您随时可以中断运行，并查看迄今为止的最佳结果。如果将 keep_training 设置为 True，则 TPOT 将从离开的位置继续训练

备注：

如果文件的目标列与默认值 ( target ) 不同，则需要手动更新target_column环境变量。您可以在“新建管道运行”窗口中执行此操作，方法是选择“输入参数”部分的“+ 新增”按钮。在“环境变量”字段中添加变量 ( target_column )，然后在“值”字段中添加文件中的列名称。

工件

TPOT 将优化管道的相应 Python 代码导出至名为 TPOT_pipeline.py 的 python 文件。代码运行完成后，TPOT_pipeline.py 文件将包含优化管道的 Python 代码。

Paper

该模型基于以下两篇论文：

• “使用功能集选取器将基于树的自动化机器学习扩展到生物医学大数据。”来自 Trang T. Le、Weixuan Fu 和 Jason H. Moore (2020)
• Randal S. Olson、Nathan Bartley、Ryan J. Urbanowicz 和 Jason H. Moore 联合发表的《Evaluation of a Tree-based Pipeline Optimization Tool for Automating Data Science》（评估用于自动化数据科学的基于树的管道优化工具）