时间戳

首页 > 解决方案 > AdaBoost - 获取特定数量的估计器的预测

prediction - AdaBoost - 获取特定数量的估计器的预测

问题描述

我使用来自sklearn.ensemble的AdaBoostClassifier。我使用 1000 个估计器训练了我的模型：

model = AdaBoostClassifier(
    base_estimator = DecisionTreeClassifier(max_depth = 6),
    n_estimators = 1000,
    learning_rate = 0.2
)
model.fit(X_train, y_train)

然后使用生成器model.staged_predict_proba(X_test)，我知道X_test数据的最佳精度是 814 个估计器。

现在我不想使用生成器model.staged_predict_proba(X_test)对新的测试数据X_test_2进行预测，因为计算每个估计器数量的预测需要很多时间（数据集真的很大）。我只想根据 814 个估计器计算模型的预测。我没有找到办法。AdaBoostClassifier可以吗？我认为应该是。

标签： predictionscikit-learnadaboost

解决方案

推荐阅读