Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
36 commits
Select commit Hold shift + click to select a range
51c4420
Fixes.
trivialfis Jul 28, 2025
5a25c14
Define categories.
trivialfis Jul 19, 2025
2d0c93b
Look into hidden methods.
trivialfis Jul 28, 2025
fe79ad8
Use device categories.
trivialfis Jul 29, 2025
ac07a88
copilot.
trivialfis Jul 29, 2025
b1763a1
maybe nvcc bug.
trivialfis Jul 29, 2025
9de6429
Work on mixed tests.
trivialfis Jul 29, 2025
bbf0bfe
Work on retrieving the cats.
trivialfis Jul 30, 2025
f927d37
fix
trivialfis Jul 30, 2025
7f98ff4
Cleanup gpu predictor.
trivialfis Jul 30, 2025
003575e
Unify the accessors.
trivialfis Jul 30, 2025
dbbe9b4
Cleanup.
trivialfis Jul 30, 2025
fa1ead1
rename.
trivialfis Jul 30, 2025
f60ddf2
Remove outdated documents.
trivialfis Jul 30, 2025
6846ca0
handle empty.
trivialfis Jul 30, 2025
f93922a
Use stream.
trivialfis Jul 30, 2025
e6dace7
Remove the container.
trivialfis Jul 30, 2025
dceb72a
macos
trivialfis Jul 30, 2025
3769972
Fix rebase.
trivialfis Jul 30, 2025
7087ae6
Guards.
trivialfis Jul 30, 2025
a5813b4
Guard.
trivialfis Jul 30, 2025
3081a7c
cross system copy.
trivialfis Jul 30, 2025
acc9223
DCats.
trivialfis Jul 30, 2025
d7be355
Cleanup.
trivialfis Jul 30, 2025
c02cbb6
typo, work with proxy.
trivialfis Jul 30, 2025
8ff81cf
race.
trivialfis Jul 30, 2025
a15fba6
Add tests for recoded prediction.
trivialfis Jul 31, 2025
8f52461
cleanup.
trivialfis Jul 31, 2025
b74f694
Error handling.
trivialfis Jul 31, 2025
72e8b14
all types.
trivialfis Jul 31, 2025
479486b
test invalid types.
trivialfis Jul 31, 2025
4bae106
CPU error handling.
trivialfis Jul 31, 2025
aeb1bbc
Cleanup.
trivialfis Jul 31, 2025
cd392f6
Update doc.
trivialfis Jul 31, 2025
268cd2c
CPU check.
trivialfis Jul 31, 2025
85bb3a5
Cleanup.
trivialfis Jul 31, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions doc/python/python_api.rst
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,8 @@ Core Data Structure
:members:
:show-inheritance:

.. autoclass:: xgboost.core.Categories

Learning API
------------
.. automodule:: xgboost.training
Expand Down
19 changes: 17 additions & 2 deletions include/xgboost/json.h
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,11 @@ class Value {
kI8Array = 9,
kU8Array = 10,
kI16Array = 11,
kI32Array = 12,
kI64Array = 13
kU16Array = 12,
kI32Array = 13,
kU32Array = 14,
kI64Array = 15,
kU64Array = 16,
};

explicit Value(ValueKind _kind) : kind_{_kind} {}
Expand Down Expand Up @@ -192,14 +195,26 @@ using U8Array = JsonTypedArray<std::uint8_t, Value::ValueKind::kU8Array>;
* @brief Typed UBJSON array for int16_t.
*/
using I16Array = JsonTypedArray<std::int16_t, Value::ValueKind::kI16Array>;
/**
* @brief Typed UBJSON array for uint16_t.
*/
using U16Array = JsonTypedArray<std::uint16_t, Value::ValueKind::kU16Array>;
/**
* @brief Typed UBJSON array for int32_t.
*/
using I32Array = JsonTypedArray<std::int32_t, Value::ValueKind::kI32Array>;
/**
* @brief Typed UBJSON array for uint32_t.
*/
using U32Array = JsonTypedArray<std::uint32_t, Value::ValueKind::kU32Array>;
/**
* @brief Typed UBJSON array for int64_t.
*/
using I64Array = JsonTypedArray<std::int64_t, Value::ValueKind::kI64Array>;
/**
* @brief Typed UBJSON array for uint64_t.
*/
using U64Array = JsonTypedArray<std::uint64_t, Value::ValueKind::kU64Array>;

class JsonObject : public Value {
public:
Expand Down
3 changes: 3 additions & 0 deletions include/xgboost/json_io.h
Original file line number Diff line number Diff line change
Expand Up @@ -143,8 +143,11 @@ class JsonWriter {
virtual void Visit(I8Array const* arr);
virtual void Visit(U8Array const* arr);
virtual void Visit(I16Array const* arr);
virtual void Visit(U16Array const* arr);
virtual void Visit(I32Array const* arr);
virtual void Visit(U32Array const* arr);
virtual void Visit(I64Array const* arr);
virtual void Visit(U64Array const* arr);
virtual void Visit(JsonObject const* obj);
virtual void Visit(JsonNumber const* num);
virtual void Visit(JsonInteger const* num);
Expand Down
2 changes: 1 addition & 1 deletion python-package/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ follow_imports = "silent"

[tool.pylint.main]
ignore = ["tests"]
extension-pkg-whitelist = ["numpy"]
extension-pkg-whitelist = ["numpy", "cuda"]
disable = [
"import-error",
"attribute-defined-outside-init",
Expand Down
Loading
Loading