Auto-Sklearn 是一种自动化机器学习框架,可以自动选择最优的模型。使用 Auto-Sklearn 可以在模型调参方面省去很多时间和精力。
以下是使用 Auto-Sklearn 选择模型的示例代码:
import autosklearn.classification
# 加载数据集
X_train, y_train, X_test, y_test = ...
# 创建 Auto-Sklearn 模型对象
automl = autosklearn.classification.AutoSklearnClassifier()
# 训练模型
automl.fit(X_train, y_train)
# 预测测试集的输出结果
y_pred = automl.predict(X_test)
# 输出最优模型
print("Auto-Sklearn chose the following model:", automl.show_models())
在上述示例代码中,调用模型对象的 show_models()
方法可以输出 Auto-Sklearn 自动选择的最优模型。