不想一个个试算法,能不能让它自己选?

手上这份数据到底该做分类还是回归、用哪个算法,我自己拿不准。同事说有个自动化的控件能省掉选型这步,是这样吗?

2 个赞

是的,自动化机器学习控件就是干这个的。把数据接进去,它会自己做特征选择、特征预处理、特征构建,还会做模型选择和参数调优,用的是遗传算法,会去探大量可能的流程和超参组合,最后给出一个合适的模型。基本配置里能调迭代次数(默认 5)、每代保留模型数(5)、单个产生后代数(3)、终止检测迭代数(3),还有个启用随机种子的开关,开了每次执行结果一致。指标选择里分类任务可选 accuracy、f1 系列这些,回归任务能选 r2 和 neg_mean_squared_error。想快速出一版基线,用它挺省事。