Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Convert labels into tensor. #7456

Merged
merged 3 commits into from Dec 16, 2021
Merged

Conversation

trivialfis
Copy link
Member

@trivialfis trivialfis commented Nov 20, 2021

  • Add a new ctor to tensor for initilizer_list.
  • Change labels from host device vector to tensor.
  • Rename the field from labels_ to labels since it's a public member. I took the opportunity to remove the NOLINT.

Rename.

Start converting labels to tensor.

Some more.

Initialization.

Fix.

Fixes.

Fix.

lint.

Fix.

Fix.

Empty check.
@trivialfis trivialfis changed the title [WIP] Convert labels into tensor. Convert labels into tensor. Dec 5, 2021
@trivialfis trivialfis marked this pull request as ready for review December 5, 2021 23:01
src/data/data.cc Show resolved Hide resolved
src/data/data.cc Show resolved Hide resolved
@trivialfis trivialfis merged commit 5b1161b into dmlc:master Dec 16, 2021
@trivialfis trivialfis deleted the tensor-labels branch December 16, 2021 16:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants