Xgboost預測predict_proba()輸出概率一致

在用Xgboost時,可以使用xgb.predict_proba()輸出各個類型的預測概率,但出現預測的各個類型概率一致的情況,你有可能是犯了以下錯誤: 模型初始化沒有注意加predictor參數,該參數默認和你使用GPU還是CPU訓練模型有關,若你使用GPU訓練模型,改參數默認爲’gpu_predictor’,假如你用cpu訓練模型,默認就爲’cpu_predictor’。當你使用GPU跑出模型
相關文章
相關標籤/搜索