Skip to content

feat: allow for gather op with private/advice indices - #454

Merged
alexander-camuto merged 9 commits into
mainfrom
ac/private-gather
Sep 4, 2023
Merged

alexander-camuto merged 9 commits into
mainfrom
ac/private-gather

Conversation

@alexander-camuto

@alexander-camuto alexander-camuto commented Sep 3, 2023 •

Copy link
Copy Markdown
Contributor
  • Allows for private / advice indices for indices in the gather op
  • This unlocks models like sklearns' decision tree -- which we convert to a nice pytorch graph using sk2torch
  • We add decision tree models from sklearn to the CI
  • The dynamic gather op also allows us to implement ops like argmax and argmin which we also add here !

Resolves #99

@alexander-camuto
alexander-camuto marked this pull request as ready for review September 4, 2023 16:21
@alexander-camuto
alexander-camuto merged commit 70c50d8 into main Sep 4, 2023
@alexander-camuto
alexander-camuto deleted the ac/private-gather branch September 4, 2023 16:30
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.

Random Forest / Decision Tree tensor op

1 participant