Inheritance diagram for xgboost.spark.data.PartIter:
Public Member Functions | |
| None | __init__ (self, Dict[str, List] data, Optional[int] device_id, **Any kwargs) |
| int | next (self, Callable input_data) |
| None | reset (self) |
 Public Member Functions inherited from xgboost.core.DataIter | |
| Tuple[Callable, Callable] | get_callbacks (self, bool allow_host, bool enable_categorical) |
| "_ProxyDMatrix" | proxy (self) |
| None | reraise (self) |
| None | __del__ (self) |
Protected Member Functions | |
| Optional[pd.DataFrame] | _fetch (self, Optional[Sequence[pd.DataFrame]] data) |
| Protected Member Functions inherited from xgboost.core.DataIter | |
| _T | _handle_exception (self, Callable fn, _T dft_ret) |
| None | _reset_wrapper (self, None this) |
| int | _next_wrapper (self, None this) |
Protected Attributes | |
| _iter | |
| _device_id | |
| _data | |
| _kwargs | |
| Protected Attributes inherited from xgboost.core.DataIter | |
| _handle | |
| _enable_categorical | |
| _allow_host | |
| _release | |
| _reset_callback | |
| _next_callback | |
| _next_wrapper | |
| _exception | |
| _temporary_data | |
| _data_ref | |
Additional Inherited Members | |
| Data Fields inherited from xgboost.core.DataIter | |
| cache_prefix | |
| proxy | |
Detailed Description
Iterator for creating Quantile DMatrix from partitions.
Constructor & Destructor Documentation
◆ __init__()
| None xgboost.spark.data.PartIter.__init__ | ( | self, | |
| Dict[str, List] | data, | ||
| Optional[int] | device_id, | ||
| **Any | kwargs | ||
| ) |
Reimplemented from xgboost.core.DataIter.
Member Function Documentation
◆ next()
| int xgboost.spark.data.PartIter.next | ( | self, | |
| Callable | input_data | ||
| ) |
Set the next batch of data.
Parameters
----------
input_data:
A function with same data fields like `data`, `label` with
`xgboost.DMatrix`.
Returns
-------
0 if there's no more batch, otherwise 1.
Reimplemented from xgboost.core.DataIter.
◆ reset()
| None xgboost.spark.data.PartIter.reset | ( | self | ) |
Reimplemented from xgboost.core.DataIter.
The documentation for this class was generated from the following file:
- External/xgboost/python-package/xgboost/spark/data.py
