EZtune:一个用于 R 语言中超参数自动调优的包
机器学习
2023-03-23 v1 数学软件
统计计算
摘要
近年来,统计学习模型日益流行。其中许多模型具有必须调优的超参数,以使模型表现良好。调优这些参数并非易事。EZtune 是一个具有简洁用户界面的 R 包,可对联支持向量机、adaboost、梯度提升机和弹性网络进行调优。我们首先简要总结 EZtune 可调优的模型,包括讨论它们各自的超参数。然后我们比较使用 EZtune、caret 和 tidymodels 的便捷性。接着比较用 EZtune 和 tidymodels 调优模型的准确率与计算时间。最后演示如何利用 EZtune 帮助选择具有最优预测能力的最终模型。我们的比较表明,EZtune 可对联支持向量机和梯度提升机进行调优,并且还提供了一个易于使用的用户界面,即使对于统计学习模型或 R 的初学者也能轻松上手。
引用
@article{arxiv.2303.12177,
title = {EZtune: A Package for Automated Hyperparameter Tuning in R},
author = {Jill Lundell},
journal= {arXiv preprint arXiv:2303.12177},
year = {2023}
}
备注
10 pages, 2 figures, submitted to the R Journal