版本 1.1.1

我们很高兴地宣布 AutoGluon 1.1.1 版本发布。

AutoGluon 1.1.1 包含了表格数据、时间序列和多模态模块的 bug 修复和日志改进,以及对 PyTorch 2.2 和 2.3 的支持。

加入社区:
获取最新动态:推特

此版本支持 Python 3.8、3.9、3.10 和 3.11 版本。不支持加载在旧版本 AutoGluon 上训练的模型。请使用 AutoGluon 1.1.1 重新训练模型。

此版本包含了来自 10 位贡献者的 52 次提交

通用

表格数据

注意:尝试加载一个在旧 AutoGluon 版本上训练的、包含 FastAI 模型的 TabularPredictor 时,调用 predict 会引发异常,这是由于 model-interals.pkl 路径的一个修复导致的。请确保版本匹配。

  • 修复当 num_gpus>0 且启用了 dynamic_stacking 时发生的死锁。 @Innixma (#4208)

  • 改进决策阈值校准。 @Innixma (#4136, #4137)

  • 改进动态 stacking 日志记录。 @Innixma (#4208, #4262)

  • 修复 LightGBM 早停时回归评估指标(RMSE 和 MSE 除外)计算不正确的问题。 @Innixma (#4174)

  • 修复 LightGBM 早停时自定义多类别评估指标计算不正确的问题。 @Innixma (#4250)

  • 修复使用 NN_TORCH 和 FASTAI 模型时 HPO(超参数优化)崩溃的问题。 @Innixma (#4232)

  • 改进 NN_TORCH 运行时估计。 @Innixma (#4247)

  • 添加推理吞吐量日志记录。 @Innixma (#4200)

  • 由于观察到性能下降,对线性模型禁用 sklearnex。 @Innixma (#4223)

  • 改进 Kaggle 中 sklearnex 的日志详细程度。 @Innixma (#4216)

  • 将缓存的版本文件重命名为 version.txt。 @Innixma (#4203)

  • 添加对线性模型的 refit_full 支持。 @Innixma (#4222)

  • 为 AsTypeFeatureGenerator 添加详细异常日志记录。 @Innixma (#4251, #4252)

时间序列

  • 确保 prediction_length 存储为整数。 @shchur (#4160)

  • 修复表格模型预处理失败的边缘情况。 @shchur (#4175)

  • 修复当 predictor 移动到不同目录时表格模型加载失败的问题。 @shchur (#4171)

  • 修复当 predictor 保存到现有 predictor 之上时缓存预测出错的问题。 @shchur (#4202)

  • 使用 Chronos 模型的 AutoGluon 分支。 @shchur (#4198)

  • 修复 Chronos 推理中的差一错误。 @canerturkmen (#4205)

  • 将缓存的版本文件重命名为 version.txt。 @Innixma (#4203)

  • 在 MLForecast 的回退模型中使用正确的目标和分位数级别。 @shchur (#4230)

多模态

文档和 CI

贡献者

完整贡献者列表(按提交次数排序)

@Innixma @shchur @Harry-zzh @suzhoum @zhiqiangdon @lovvge @rey-allan @prateekdesai04 @canerturkmen @FANGAreNotGnu

新贡献者

  • @lovvge 在 https://github.com/autogluon/autogluon/commit/57a15fcfbbbc94514ff20ed2774cd447d9f4115f 贡献了他们的第一个提交

  • @rey-allan#4145 贡献了他们的第一个提交