Saya menjalankan xgboost di spark dan bertemu AttributeError: 'XGBoostClassifier' object has no attribute 'booster'

def train_model(trainDF):
    xgboost = XGBoostClassifier(
        featuresCol="features",
        labelCol="label",
        predictionCol="prediction",
        objective='multi:softprob',
        numClass=10,
        missing=0.0
    )
    pipeline = Pipeline(stages=[xgboost])
    model = pipeline.fit(trainDF)
    featureScoreMap = xgb_model.booster.getFeatureScore()

Bagaimana cara mendapatkan pentingnya fitur xgboost4j?

0
rosefun 7 Juli 2020, 09:50

1 menjawab

Jawaban Terbaik

Coba ini- Dapatkan fitur penting dari model pipa yang memiliki model xgboost sebagai tahap pertama

Di Scala

val xgboostModel = model.stages.apply(0).asInstanceOf[XGBoostClassificationModel]

xgboostModel.nativeBooster.getFeatureScore()

Dengan Python (dari komentar)

model.stages[0].nativeBooster.getScore("", "gain")
2
Som 9 Juli 2020, 06:11