AutoML手动选择模型

Question

这是一段时间以来，我正在寻找最好的管道做一些分类使用AutoML。但是我想知道是否可以手动选择模型，然后优化它的超参数。例如，我只想优化SVM的超参数，而不关心其他模型。

Answer 1

您只能在MLJAR AutoML中优化所选模型。它是开放源代码的AutoML，代码可在GitHub：https://github.com/mljar/mljar-supervised上找到。

示例代码如下所示：

automl = AutoML(algorithms=["Xgboost"], mode="Compete")
automl.fit(X, y)

上面的代码将只调优Xgboost算法。需要模式Compete，因为MLJAR AutoML可以在三种模式下工作：Explain、Perform和Compete。在MLJAR AutoML中可用的算法:基线、线性、随机森林、额外树、决策树、神经网络、最近邻、Xgboost、LightGBM、CatBoost。

我是MLJAR AutoML的作者，我很乐意帮助您设置它并运行。