bartz 0.5.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (13) hide show
  1. bartz/BART.py +196 -103
  2. bartz/__init__.py +1 -1
  3. bartz/_version.py +1 -1
  4. bartz/debug.py +1 -1
  5. bartz/grove.py +43 -2
  6. bartz/jaxext.py +82 -33
  7. bartz/mcmcloop.py +367 -114
  8. bartz/mcmcstep.py +1322 -807
  9. bartz/prepcovars.py +3 -1
  10. {bartz-0.5.0.dist-info → bartz-0.6.0.dist-info}/METADATA +7 -5
  11. bartz-0.6.0.dist-info/RECORD +13 -0
  12. {bartz-0.5.0.dist-info → bartz-0.6.0.dist-info}/WHEEL +1 -1
  13. bartz-0.5.0.dist-info/RECORD +0 -13
bartz/prepcovars.py CHANGED
@@ -22,6 +22,8 @@
22
22
  # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
23
23
  # SOFTWARE.
24
24
 
25
+ """Functions to preprocess data."""
26
+
25
27
  import functools
26
28
 
27
29
  import jax
@@ -55,9 +57,9 @@ def quantilized_splits_from_matrix(X, max_bins):
55
57
  """
56
58
  out_length = min(max_bins, X.shape[1]) - 1
57
59
 
58
- # return _quantilized_splits_from_matrix(X, out_length)
59
60
  @functools.partial(jaxext.autobatch, max_io_nbytes=2**29)
60
61
  def quantilize(X):
62
+ # wrap this function because autobatch needs traceable args
61
63
  return _quantilized_splits_from_matrix(X, out_length)
62
64
 
63
65
  return quantilize(X)
{bartz-0.5.0.dist-info → bartz-0.6.0.dist-info}/METADATA RENAMED
@@ -1,14 +1,16 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: bartz
3
- Version: 0.5.0
3
+ Version: 0.6.0
4
4
  Summary: Super-fast BART (Bayesian Additive Regression Trees) in Python
5
5
  Author: Giacomo Petrillo
6
6
  Author-email: Giacomo Petrillo <info@giacomopetrillo.com>
7
7
  License-Expression: MIT
8
- Requires-Dist: jax>=0.4.35,<1
9
- Requires-Dist: jaxlib>=0.4.35,<1
10
- Requires-Dist: numpy>=1.25.2,<3
11
- Requires-Dist: scipy>=1.11.4,<2
8
+ Requires-Dist: equinox>=0.12.2
9
+ Requires-Dist: jax>=0.4.35
10
+ Requires-Dist: jaxlib>=0.4.35
11
+ Requires-Dist: jaxtyping>=0.3.2
12
+ Requires-Dist: numpy>=1.25.2
13
+ Requires-Dist: scipy>=1.11.4
12
14
  Requires-Python: >=3.10
13
15
  Project-URL: Documentation, https://gattocrucco.github.io/bartz/docs-dev
14
16
  Project-URL: Homepage, https://github.com/Gattocrucco/bartz
bartz-0.6.0.dist-info/RECORD ADDED
@@ -0,0 +1,13 @@
1
+ bartz/.DS_Store,sha256=7191af46d7b8c0d4c03c502f94eb01353bc2e615d75c45b3af0e31ab238034b5,6148
2
+ bartz/BART.py,sha256=4de980d640f694e4b65fb17d7c5fe605e15c10f730ee842d39e45501ae8b6829,21606
3
+ bartz/__init__.py,sha256=b57b8ecb9f97ced8a543fa65e191ebda4addd79ef12a8fb357e4f6b23ac2ca72,1434
4
+ bartz/_version.py,sha256=0816378ec0bef470a6ede67a08a0feb182c27a3a8e275a5858f40ce0b1885dc2,22
5
+ bartz/debug.py,sha256=6e06cccfe1c5d6858e62200cd679b73a528af27780d33ba6c164a3a6d96660a5,5356
6
+ bartz/grove.py,sha256=955f03269f373e7bc4f917c6d2ba598e363e8f910adea7f2195693c8443ad381,9636
7
+ bartz/jaxext.py,sha256=ac9aff86e4fb9ff43b5d98995df2b9416a715c82f87270eb1c547fae8973c6dd,13122
8
+ bartz/mcmcloop.py,sha256=e2ad8ae5fb0a49ad3c874e25b6bf2ea966f4d678c0b99bf575393cbe388514bf,17695
9
+ bartz/mcmcstep.py,sha256=5ef4303dfe06be65c3bdfaf44217c74c3e6244444b03dcc5d6eacdee8c9b1d34,72891
10
+ bartz/prepcovars.py,sha256=0db7601a7a73cba508ed4d8c1ca9ac07baacd4f756978bbc80cf173c9257a5ed,5894
11
+ bartz-0.6.0.dist-info/WHEEL,sha256=ec89e25a684ec3ad8fb2a9be806e47666a27f7dcdcf9b55bff3f1d3ed5e60e0a,78
12
+ bartz-0.6.0.dist-info/METADATA,sha256=c35ffe8f0dab8de368e277041105f64650b26f9f4582015cb1b334f88a0e72d6,2846
13
+ bartz-0.6.0.dist-info/RECORD,,
{bartz-0.5.0.dist-info → bartz-0.6.0.dist-info}/WHEEL RENAMED
@@ -1,4 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: uv 0.7.4
2
+ Generator: uv 0.7.8
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
bartz-0.5.0.dist-info/RECORD DELETED
@@ -1,13 +0,0 @@
1
- bartz/.DS_Store,sha256=7191af46d7b8c0d4c03c502f94eb01353bc2e615d75c45b3af0e31ab238034b5,6148
2
- bartz/BART.py,sha256=50faa27d82bc1c5ab6eb676a006bd7a95e2cbe10a541c5de7a100f068a76a820,17898
3
- bartz/__init__.py,sha256=c3b1d04b62e52220f060d28f38403a536fcbe2cea06d0a4087dc0143b05cb1fd,1433
4
- bartz/_version.py,sha256=d0f688d9e48e0a9e6490d7292a95126cb1dfebaacbf714331696322c6a44b723,22
5
- bartz/debug.py,sha256=0ab2b06e53245b763d4c684653031c912ed794e328d2f427dbfb017c975840af,5359
6
- bartz/grove.py,sha256=13d05ca6877b073cfc2d0e887fc03f74be35eadb6b587b83a0a0d41de1accb10,8494
7
- bartz/jaxext.py,sha256=a439363fd44be9a26770a65dd30c3d140a877ec0ddcdd65ba8fed4471c629cd0,11552
8
- bartz/mcmcloop.py,sha256=15f7de40d1b3a57183f90db2972b66ab7b455f4829cca22c5686736b154c1fac,9216
9
- bartz/mcmcstep.py,sha256=48aaf9d4a32aa4b84a73b4ec365fe712204dd872e5df3de5cc3715a37e362430,60504
10
- bartz/prepcovars.py,sha256=e76c05c5218a47096c368f28c2df0178b44b428d26ba9c15fa6dd6c09fd2f2a1,5849
11
- bartz-0.5.0.dist-info/WHEEL,sha256=11319513b72fe7d4379f6f9f7984b3cca85f688201c09325839a04908e107b75,78
12
- bartz-0.5.0.dist-info/METADATA,sha256=d875077205f7d03c4c4f8c84acfe3079c4808be32ea3ab358640fe581b6f3b46,2795
13
- bartz-0.5.0.dist-info/RECORD,,