Как указать несколько столбцов в xgboost.trainWithDataframe при использовании искры?

Question

введите описание изображения здесь

Это API-документ, представленный на xgboost.com, кажется, что я могу установить только один столбец как "featureCol".

Answer 1

Как и в случае любого другого ML Estimator в Spark, ожидается, что inputCol будет Vector собранных функций.Перед применением Estimator вы должны использовать инструменты из org.apache.spark.ml.feature для извлечения, преобразования и сборки вектора объектов.

Вы можете проверить Как векторизовать столбцы DataFrame для алгоритмов ML? дляпример Pipeline.