Skip to content

Conversation

@philsmt
Copy link
Contributor

@philsmt philsmt commented Dec 19, 2025

When using data with more than one entry per train, I find it sometimes convenient (and also regularly more performant) to use plain old ndarrays and have arrays lying around indicating the first and last index of a given train.

This PR adds a method KeyData.train_index_bounds() to deliver this based on KeyData.data_counts(). I went with the traditional labelled approach, though I don't have hard feelings for that.

@philsmt philsmt requested a review from takluyver December 19, 2025 10:41
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