【发布时间】:2020-09-29 12:57:39
【问题描述】:
我有一个数据来训练模型。此外,我还有另一个数据可以每周测试模型的性能。但是,模型似乎并不稳定。训练分数和每周测试分数之间存在一些差异。另一方面,这是一个欺诈问题,我正在使用 XGBoosting 方法。如何使模型稳定?我可以使用不同的算法和参数。
parameters = {
'n_estimators':[100],
'max_depth':[5],
'learning_rate':[0.1],
'classifier__min_sample_leaf':[5],
'classifier__criterion':['gini']
}
xgboost = XGBClassifier(scale_pos_weight=30)
xgboost_gs = GridSearchCV(xgboost, parameters, scoring='recall', cv=5, verbose=False)
xgboost_gs.fit(X_train, y_train)
【问题讨论】:
-
你可以试试keras或tensorflow之类的深度学习方法。您可以使用此设置检查它是否正常工作kaggle.com/dakshmiglani/…
标签: python machine-learning data-science random-forest xgboost