WeichenXu

Results 53 issues of WeichenXu

### Willingness to contribute Yes. I would be willing to contribute this feature with guidance from the MLflow community. ### Proposal Summary Example: `run_id in a_list and tag.xx == yy`,...

enhancement
area/tracking

### Description Add config option for Ray Node Runtime to specify visible cuda devices on a node: Suppose on a machine, there are 4 GPUs, and GPU-0/ GPU-1 are used...

enhancement

Closes https://github.com/dmlc/xgboost/issues/8108 Make Xgboost estimator support using sparse matrix as optimization.

Xgboost spark: Support XGBRanker

feature-request

Xgboost pyspark: Support pyspark Sparse Vector

feature-request

Xgboost pyspark: Support external memory mode

feature-request
external-memory

Apply dmatrix iteration iterface in PySpark xgboost and support external memory mode.

Signed-off-by: Weichen Xu Upgrade xgboost dependency to be 1.6.1 version

org.apache.spark.ml.bundle.tree.decision.SparkNodeWrapper only support "gini" impurity ("gini" is hardcoded), we should update code to support all kinds of impurity. Currently, any type node in mleap will be regarded as "gini" impurity.

NaiveBayes model support Gauss type. We need update code in `ml.combust.mleap.core.classification.NaiveBayesModel` to make it also support Gauss type.