代码之家  ›  专栏  ›  技术社区  ›  Venkatachalam

Xgboost随机种子设定不允许繁殖能力

  •  0
  • Venkatachalam  · 技术社区  · 7 年前

    尽管为 random_state 和/或 seed 参数时,性能不可复制 Xgboost Sklearn API包装器

    这是密码

    from xgboost.sklearn import XGBClassifier
    from sklearn.metrics import roc_auc_score
    
    (X_train,y_train),(X_eval,y_eval) = pickle.load(open('xxxx.pkl',"rb"))
    
    hyperparams = {'eval_metric': 'auc', 'colsample_bylevel': 0.7, 'learning_rate': 0.125,  'random_state': 0}
    GBM = XGBClassifier(**hyperparams)
    GBM.fit(X_train,y_train,eval_metric="auc",verbose = True,eval_set=[(X_eval,y_eval)],early_stopping_rounds=2)
    
    print(roc_auc_score(y_eval, GBM.predict_proba(X_eval)[:,1]))
    

    每次运行上述代码段时,一致性值都不同。

    78.9451246
    79.001542
    

    一些参考资料 issues

    0 回复  |  直到 5 年前