python - TPOT 训练时间太长

Question

我第一次尝试在大约 7000 行的数据集上使用 tpot，当尝试在占整个数据集 25% 的训练数据集上训练 tpot 时，tpot 花费的时间太长。我在 google colab 上运行代码大约 45 分钟，优化进度仍为 4%。我一直在尝试使用以下示例：http: //epistasislab.github.io/tpot/examples/。tpot 需要这么长时间是典型的吗，因为到目前为止我认为它甚至不值得尝试使用它

标签： pythontpot

Answer 1

TPOT 可能需要很长时间，具体取决于您拥有的数据集。您必须考虑 TPOT 正在做什么：TPOT 正在评估数千个分析管道并在后台在您的数据集上拟合数千个 ML 模型，如果您有一个大数据集，那么所有这些拟合可能需要很长时间——尤其是如果您在功能较弱的计算机上运行它。

如果您想要更快的结果，您有几个选择：

1. 使用“TPOT light”配置，使用更简单的模型，运行速度更快。

2. 将n_jobs参数设置为-1或大于 的数字1，这将允许 TPOT 并行评估管道。-1如果您有一台多核机器，它将使用所有可用的内核并显着加快速度。

3. subsample使用参数对数据进行二次采样。默认值为 1.0，对应于使用 100% 的训练数据。您可以对较低百分比的数据进行二次抽样，TPOT 将运行得更快。