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

Support gpu version crossvalidator for LinearRegression #319

Merged
merged 2 commits into from
Jul 4, 2023

Conversation

wbo4958
Copy link
Collaborator

@wbo4958 wbo4958 commented Jul 3, 2023

No description provided.

Signed-off-by: Bobby Wang <wbo4958@gmail.com>
Signed-off-by: Bobby Wang <wbo4958@gmail.com>
@wbo4958
Copy link
Collaborator Author

wbo4958 commented Jul 3, 2023

build

Copy link
Collaborator

@leewyang leewyang left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

One question/comment, but otherwise LGTM

coef_: List[float],
intercept_: float,
coef_: Union[List[float], List[List[float]]],
intercept_: Union[float, List[float]],
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Wondering if just assuming lists here would simplify things elsewhere, where the single model is just represented as a list of one item.

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Wow, Good suggestion. I will file a followup PR to handle it not only for LinearRegression but also for RandomForest.

@wbo4958 wbo4958 merged commit 326520b into NVIDIA:branch-23.06 Jul 4, 2023
1 check passed
@wbo4958 wbo4958 deleted the lr-cv branch July 4, 2023 01:12
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.

2 participants