Set meta info for DMatrix. See doc string for :py:obj:`xgboost.DMatrix`.
(
self,
*,
label: Optional[ArrayLike] = None,
weight: Optional[ArrayLike] = None,
base_margin: Optional[ArrayLike] = None,
group: Optional[ArrayLike] = None,
qid: Optional[ArrayLike] = None,
label_lower_bound: Optional[ArrayLike] = None,
label_upper_bound: Optional[ArrayLike] = None,
feature_names: Optional[FeatureNames] = None,
feature_types: Optional[FeatureTypes] = None,
feature_weights: Optional[ArrayLike] = None,
)
| 856 | |
| 857 | @_deprecate_positional_args |
| 858 | def set_info( |
| 859 | self, |
| 860 | *, |
| 861 | label: Optional[ArrayLike] = None, |
| 862 | weight: Optional[ArrayLike] = None, |
| 863 | base_margin: Optional[ArrayLike] = None, |
| 864 | group: Optional[ArrayLike] = None, |
| 865 | qid: Optional[ArrayLike] = None, |
| 866 | label_lower_bound: Optional[ArrayLike] = None, |
| 867 | label_upper_bound: Optional[ArrayLike] = None, |
| 868 | feature_names: Optional[FeatureNames] = None, |
| 869 | feature_types: Optional[FeatureTypes] = None, |
| 870 | feature_weights: Optional[ArrayLike] = None, |
| 871 | ) -> None: |
| 872 | """Set meta info for DMatrix. See doc string for :py:obj:`xgboost.DMatrix`.""" |
| 873 | from .data import dispatch_meta_backend |
| 874 | |
| 875 | if label is not None: |
| 876 | self.set_label(label) |
| 877 | if weight is not None: |
| 878 | self.set_weight(weight) |
| 879 | if base_margin is not None: |
| 880 | self.set_base_margin(base_margin) |
| 881 | if group is not None: |
| 882 | self.set_group(group) |
| 883 | if qid is not None: |
| 884 | self.set_uint_info("qid", qid) |
| 885 | if label_lower_bound is not None: |
| 886 | self.set_float_info("label_lower_bound", label_lower_bound) |
| 887 | if label_upper_bound is not None: |
| 888 | self.set_float_info("label_upper_bound", label_upper_bound) |
| 889 | if feature_names is not None: |
| 890 | self.feature_names = feature_names |
| 891 | if feature_types is not None: |
| 892 | self.feature_types = feature_types |
| 893 | if feature_weights is not None: |
| 894 | dispatch_meta_backend( |
| 895 | matrix=self, data=feature_weights, name="feature_weights" |
| 896 | ) |
| 897 | |
| 898 | def _get_info(self, field: str) -> NumpyOrCupy: |
| 899 | """Get meta info.""" |