Skip to content

Allow empty datasets in survival:aft objective and aft-nloglik metric #6139

Closed
@hcho3

Description

@hcho3

Currently, survival:aft objective and aft-nloglik metric throws an error when an empty data matrix is passed in. Empty data matrix is a common occurrence when XGBoost is used with Dask, due to quirks in how Dask partitions the data matrix.

The AFT objective and metric should handle empty data matrices like other element-wise objective and metrics, such as reg:squarederror and logloss.

cc @trivialfis

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions