pkboost 0.1.0__tar.gz

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.

Potentially problematic release.

This version of pkboost might be problematic. Click here for more details.

Files changed (47) hide show
  1. pkboost-0.1.0/.gitignore +40 -0
  2. pkboost-0.1.0/Cargo.lock +538 -0
  3. pkboost-0.1.0/Cargo.toml +46 -0
  4. pkboost-0.1.0/PKG-INFO +370 -0
  5. pkboost-0.1.0/README.md +351 -0
  6. pkboost-0.1.0/benchmark results/1.png +0 -0
  7. pkboost-0.1.0/benchmark results/all_benchmarks_results.csv +5 -0
  8. pkboost-0.1.0/benchmark results/wosconsin.png +0 -0
  9. pkboost-0.1.0/create_bigger_test.py +28 -0
  10. pkboost-0.1.0/create_extreme_imbalance.py +38 -0
  11. pkboost-0.1.0/create_small_test.py +30 -0
  12. pkboost-0.1.0/data/preprocessor.pkl +0 -0
  13. pkboost-0.1.0/diagrams/pkboost_system_architecture.drawio +244 -0
  14. pkboost-0.1.0/diagrams/state_machine.drawio +204 -0
  15. pkboost-0.1.0/download_creditcard.py +21 -0
  16. pkboost-0.1.0/drift_comparison_all.py +141 -0
  17. pkboost-0.1.0/prepare_data.py +155 -0
  18. pkboost-0.1.0/pyproject.toml +30 -0
  19. pkboost-0.1.0/python/README.md +250 -0
  20. pkboost-0.1.0/python/example.py +49 -0
  21. pkboost-0.1.0/python/example_creditcard.py +75 -0
  22. pkboost-0.1.0/python/example_creditcard_drift.py +124 -0
  23. pkboost-0.1.0/python/example_drift.py +114 -0
  24. pkboost-0.1.0/resources/comprehensive_benchmark.py +240 -0
  25. pkboost-0.1.0/resources/pybenchmark.py +378 -0
  26. pkboost-0.1.0/resources/test_drift_comparison.py +238 -0
  27. pkboost-0.1.0/resources/text.py +110 -0
  28. pkboost-0.1.0/run_all_benchmarks.py +400 -0
  29. pkboost-0.1.0/src/adaptive_parallel.rs +161 -0
  30. pkboost-0.1.0/src/adversarial.rs +91 -0
  31. pkboost-0.1.0/src/auto_params.rs +75 -0
  32. pkboost-0.1.0/src/auto_tuner.rs +126 -0
  33. pkboost-0.1.0/src/bin/benchmark.rs +180 -0
  34. pkboost-0.1.0/src/bin/test_drift.rs +324 -0
  35. pkboost-0.1.0/src/bin/test_living.rs +67 -0
  36. pkboost-0.1.0/src/bin/test_retrain.rs +171 -0
  37. pkboost-0.1.0/src/bin/test_static.rs +129 -0
  38. pkboost-0.1.0/src/histogram_builder.rs +185 -0
  39. pkboost-0.1.0/src/lib.rs +28 -0
  40. pkboost-0.1.0/src/living_booster.rs +583 -0
  41. pkboost-0.1.0/src/loss.rs +64 -0
  42. pkboost-0.1.0/src/metabolism.rs +115 -0
  43. pkboost-0.1.0/src/metrics.rs +137 -0
  44. pkboost-0.1.0/src/model.rs +618 -0
  45. pkboost-0.1.0/src/optimized_data.rs +137 -0
  46. pkboost-0.1.0/src/python_bindings.rs +357 -0
  47. pkboost-0.1.0/src/tree.rs +514 -0
pkboost-0.1.0/.gitignore ADDED
@@ -0,0 +1,40 @@
1
+ # Rust
2
+ /target/
3
+ **/*.rs.bk
4
+ *.pdb
5
+ Cargo.lock
6
+
7
+ # IDE
8
+ .vscode/
9
+ .idea/
10
+ *.swp
11
+ *.swo
12
+ *~
13
+
14
+ # OS
15
+ .DS_Store
16
+ Thumbs.db
17
+
18
+ # Python
19
+ __pycache__/
20
+ *.py[cod]
21
+ *$py.class
22
+ *.so
23
+ .Python
24
+ env/
25
+ venv/
26
+ *.egg-info/
27
+ .pytest_cache/
28
+
29
+ # Data files (optional - uncomment if you want to exclude large datasets)
30
+ # data/*.csv
31
+
32
+ # Output files
33
+ *.png
34
+ *.csv
35
+ alb_metrics.csv
36
+ drift_comparison_*.png
37
+ real_world_benchmark_results.csv
38
+
39
+ # Logs
40
+ *.log
pkboost-0.1.0/Cargo.lock ADDED
@@ -0,0 +1,538 @@
1
+ # This file is automatically @generated by Cargo.
2
+ # It is not intended for manual editing.
3
+ version = 4
4
+
5
+ [[package]]
6
+ name = "autocfg"
7
+ version = "1.5.0"
8
+ source = "registry+https://github.com/rust-lang/crates.io-index"
9
+ checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
10
+
11
+ [[package]]
12
+ name = "cc"
13
+ version = "1.2.39"
14
+ source = "registry+https://github.com/rust-lang/crates.io-index"
15
+ checksum = "e1354349954c6fc9cb0deab020f27f783cf0b604e8bb754dc4658ecf0d29c35f"
16
+ dependencies = [
17
+ "find-msvc-tools",
18
+ "shlex",
19
+ ]
20
+
21
+ [[package]]
22
+ name = "cfg-if"
23
+ version = "1.0.3"
24
+ source = "registry+https://github.com/rust-lang/crates.io-index"
25
+ checksum = "2fd1289c04a9ea8cb22300a459a72a385d7c73d3259e2ed7dcb2af674838cfa9"
26
+
27
+ [[package]]
28
+ name = "crossbeam-deque"
29
+ version = "0.8.6"
30
+ source = "registry+https://github.com/rust-lang/crates.io-index"
31
+ checksum = "9dd111b7b7f7d55b72c0a6ae361660ee5853c9af73f70c3c2ef6858b950e2e51"
32
+ dependencies = [
33
+ "crossbeam-epoch",
34
+ "crossbeam-utils",
35
+ ]
36
+
37
+ [[package]]
38
+ name = "crossbeam-epoch"
39
+ version = "0.9.18"
40
+ source = "registry+https://github.com/rust-lang/crates.io-index"
41
+ checksum = "5b82ac4a3c2ca9c3460964f020e1402edd5753411d7737aa39c3714ad1b5420e"
42
+ dependencies = [
43
+ "crossbeam-utils",
44
+ ]
45
+
46
+ [[package]]
47
+ name = "crossbeam-utils"
48
+ version = "0.8.21"
49
+ source = "registry+https://github.com/rust-lang/crates.io-index"
50
+ checksum = "d0a5c400df2834b80a4c3327b3aad3a4c4cd4de0629063962b03235697506a28"
51
+
52
+ [[package]]
53
+ name = "csv"
54
+ version = "1.3.1"
55
+ source = "registry+https://github.com/rust-lang/crates.io-index"
56
+ checksum = "acdc4883a9c96732e4733212c01447ebd805833b7275a73ca3ee080fd77afdaf"
57
+ dependencies = [
58
+ "csv-core",
59
+ "itoa",
60
+ "ryu",
61
+ "serde",
62
+ ]
63
+
64
+ [[package]]
65
+ name = "csv-core"
66
+ version = "0.1.12"
67
+ source = "registry+https://github.com/rust-lang/crates.io-index"
68
+ checksum = "7d02f3b0da4c6504f86e9cd789d8dbafab48c2321be74e9987593de5a894d93d"
69
+ dependencies = [
70
+ "memchr",
71
+ ]
72
+
73
+ [[package]]
74
+ name = "either"
75
+ version = "1.15.0"
76
+ source = "registry+https://github.com/rust-lang/crates.io-index"
77
+ checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
78
+
79
+ [[package]]
80
+ name = "find-msvc-tools"
81
+ version = "0.1.2"
82
+ source = "registry+https://github.com/rust-lang/crates.io-index"
83
+ checksum = "1ced73b1dacfc750a6db6c0a0c3a3853c8b41997e2e2c563dc90804ae6867959"
84
+
85
+ [[package]]
86
+ name = "getrandom"
87
+ version = "0.2.16"
88
+ source = "registry+https://github.com/rust-lang/crates.io-index"
89
+ checksum = "335ff9f135e4384c8150d6f27c6daed433577f86b4750418338c01a1a2528592"
90
+ dependencies = [
91
+ "cfg-if",
92
+ "libc",
93
+ "wasi",
94
+ ]
95
+
96
+ [[package]]
97
+ name = "heck"
98
+ version = "0.5.0"
99
+ source = "registry+https://github.com/rust-lang/crates.io-index"
100
+ checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
101
+
102
+ [[package]]
103
+ name = "hermit-abi"
104
+ version = "0.5.2"
105
+ source = "registry+https://github.com/rust-lang/crates.io-index"
106
+ checksum = "fc0fef456e4baa96da950455cd02c081ca953b141298e41db3fc7e36b1da849c"
107
+
108
+ [[package]]
109
+ name = "indoc"
110
+ version = "2.0.7"
111
+ source = "registry+https://github.com/rust-lang/crates.io-index"
112
+ checksum = "79cf5c93f93228cf8efb3ba362535fb11199ac548a09ce117c9b1adc3030d706"
113
+ dependencies = [
114
+ "rustversion",
115
+ ]
116
+
117
+ [[package]]
118
+ name = "itoa"
119
+ version = "1.0.15"
120
+ source = "registry+https://github.com/rust-lang/crates.io-index"
121
+ checksum = "4a5f13b858c8d314ee3e8f639011f7ccefe71f97f96e50151fb991f267928e2c"
122
+
123
+ [[package]]
124
+ name = "libc"
125
+ version = "0.2.176"
126
+ source = "registry+https://github.com/rust-lang/crates.io-index"
127
+ checksum = "58f929b4d672ea937a23a1ab494143d968337a5f47e56d0815df1e0890ddf174"
128
+
129
+ [[package]]
130
+ name = "libmimalloc-sys"
131
+ version = "0.1.44"
132
+ source = "registry+https://github.com/rust-lang/crates.io-index"
133
+ checksum = "667f4fec20f29dfc6bc7357c582d91796c169ad7e2fce709468aefeb2c099870"
134
+ dependencies = [
135
+ "cc",
136
+ "libc",
137
+ ]
138
+
139
+ [[package]]
140
+ name = "matrixmultiply"
141
+ version = "0.3.10"
142
+ source = "registry+https://github.com/rust-lang/crates.io-index"
143
+ checksum = "a06de3016e9fae57a36fd14dba131fccf49f74b40b7fbdb472f96e361ec71a08"
144
+ dependencies = [
145
+ "autocfg",
146
+ "rawpointer",
147
+ ]
148
+
149
+ [[package]]
150
+ name = "memchr"
151
+ version = "2.7.6"
152
+ source = "registry+https://github.com/rust-lang/crates.io-index"
153
+ checksum = "f52b00d39961fc5b2736ea853c9cc86238e165017a493d1d5c8eac6bdc4cc273"
154
+
155
+ [[package]]
156
+ name = "memoffset"
157
+ version = "0.9.1"
158
+ source = "registry+https://github.com/rust-lang/crates.io-index"
159
+ checksum = "488016bfae457b036d996092f6cb448677611ce4449e970ceaf42695203f218a"
160
+ dependencies = [
161
+ "autocfg",
162
+ ]
163
+
164
+ [[package]]
165
+ name = "mimalloc"
166
+ version = "0.1.48"
167
+ source = "registry+https://github.com/rust-lang/crates.io-index"
168
+ checksum = "e1ee66a4b64c74f4ef288bcbb9192ad9c3feaad75193129ac8509af543894fd8"
169
+ dependencies = [
170
+ "libmimalloc-sys",
171
+ ]
172
+
173
+ [[package]]
174
+ name = "ndarray"
175
+ version = "0.16.1"
176
+ source = "registry+https://github.com/rust-lang/crates.io-index"
177
+ checksum = "882ed72dce9365842bf196bdeedf5055305f11fc8c03dee7bb0194a6cad34841"
178
+ dependencies = [
179
+ "matrixmultiply",
180
+ "num-complex",
181
+ "num-integer",
182
+ "num-traits",
183
+ "portable-atomic",
184
+ "portable-atomic-util",
185
+ "rawpointer",
186
+ "rayon",
187
+ ]
188
+
189
+ [[package]]
190
+ name = "num-complex"
191
+ version = "0.4.6"
192
+ source = "registry+https://github.com/rust-lang/crates.io-index"
193
+ checksum = "73f88a1307638156682bada9d7604135552957b7818057dcef22705b4d509495"
194
+ dependencies = [
195
+ "num-traits",
196
+ ]
197
+
198
+ [[package]]
199
+ name = "num-integer"
200
+ version = "0.1.46"
201
+ source = "registry+https://github.com/rust-lang/crates.io-index"
202
+ checksum = "7969661fd2958a5cb096e56c8e1ad0444ac2bbcd0061bd28660485a44879858f"
203
+ dependencies = [
204
+ "num-traits",
205
+ ]
206
+
207
+ [[package]]
208
+ name = "num-traits"
209
+ version = "0.2.19"
210
+ source = "registry+https://github.com/rust-lang/crates.io-index"
211
+ checksum = "071dfc062690e90b734c0b2273ce72ad0ffa95f0c74596bc250dcfd960262841"
212
+ dependencies = [
213
+ "autocfg",
214
+ ]
215
+
216
+ [[package]]
217
+ name = "num_cpus"
218
+ version = "1.17.0"
219
+ source = "registry+https://github.com/rust-lang/crates.io-index"
220
+ checksum = "91df4bbde75afed763b708b7eee1e8e7651e02d97f6d5dd763e89367e957b23b"
221
+ dependencies = [
222
+ "hermit-abi",
223
+ "libc",
224
+ ]
225
+
226
+ [[package]]
227
+ name = "numpy"
228
+ version = "0.22.1"
229
+ source = "registry+https://github.com/rust-lang/crates.io-index"
230
+ checksum = "edb929bc0da91a4d85ed6c0a84deaa53d411abfb387fc271124f91bf6b89f14e"
231
+ dependencies = [
232
+ "libc",
233
+ "ndarray",
234
+ "num-complex",
235
+ "num-integer",
236
+ "num-traits",
237
+ "pyo3",
238
+ "rustc-hash",
239
+ ]
240
+
241
+ [[package]]
242
+ name = "once_cell"
243
+ version = "1.21.3"
244
+ source = "registry+https://github.com/rust-lang/crates.io-index"
245
+ checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
246
+
247
+ [[package]]
248
+ name = "pkboost"
249
+ version = "0.1.0"
250
+ dependencies = [
251
+ "csv",
252
+ "mimalloc",
253
+ "ndarray",
254
+ "num_cpus",
255
+ "numpy",
256
+ "pyo3",
257
+ "rand",
258
+ "rayon",
259
+ "simsimd",
260
+ ]
261
+
262
+ [[package]]
263
+ name = "portable-atomic"
264
+ version = "1.11.1"
265
+ source = "registry+https://github.com/rust-lang/crates.io-index"
266
+ checksum = "f84267b20a16ea918e43c6a88433c2d54fa145c92a811b5b047ccbe153674483"
267
+
268
+ [[package]]
269
+ name = "portable-atomic-util"
270
+ version = "0.2.4"
271
+ source = "registry+https://github.com/rust-lang/crates.io-index"
272
+ checksum = "d8a2f0d8d040d7848a709caf78912debcc3f33ee4b3cac47d73d1e1069e83507"
273
+ dependencies = [
274
+ "portable-atomic",
275
+ ]
276
+
277
+ [[package]]
278
+ name = "ppv-lite86"
279
+ version = "0.2.21"
280
+ source = "registry+https://github.com/rust-lang/crates.io-index"
281
+ checksum = "85eae3c4ed2f50dcfe72643da4befc30deadb458a9b590d720cde2f2b1e97da9"
282
+ dependencies = [
283
+ "zerocopy",
284
+ ]
285
+
286
+ [[package]]
287
+ name = "proc-macro2"
288
+ version = "1.0.101"
289
+ source = "registry+https://github.com/rust-lang/crates.io-index"
290
+ checksum = "89ae43fd86e4158d6db51ad8e2b80f313af9cc74f5c0e03ccb87de09998732de"
291
+ dependencies = [
292
+ "unicode-ident",
293
+ ]
294
+
295
+ [[package]]
296
+ name = "pyo3"
297
+ version = "0.22.6"
298
+ source = "registry+https://github.com/rust-lang/crates.io-index"
299
+ checksum = "f402062616ab18202ae8319da13fa4279883a2b8a9d9f83f20dbade813ce1884"
300
+ dependencies = [
301
+ "cfg-if",
302
+ "indoc",
303
+ "libc",
304
+ "memoffset",
305
+ "once_cell",
306
+ "portable-atomic",
307
+ "pyo3-build-config",
308
+ "pyo3-ffi",
309
+ "pyo3-macros",
310
+ "unindent",
311
+ ]
312
+
313
+ [[package]]
314
+ name = "pyo3-build-config"
315
+ version = "0.22.6"
316
+ source = "registry+https://github.com/rust-lang/crates.io-index"
317
+ checksum = "b14b5775b5ff446dd1056212d778012cbe8a0fbffd368029fd9e25b514479c38"
318
+ dependencies = [
319
+ "once_cell",
320
+ "target-lexicon",
321
+ ]
322
+
323
+ [[package]]
324
+ name = "pyo3-ffi"
325
+ version = "0.22.6"
326
+ source = "registry+https://github.com/rust-lang/crates.io-index"
327
+ checksum = "9ab5bcf04a2cdcbb50c7d6105de943f543f9ed92af55818fd17b660390fc8636"
328
+ dependencies = [
329
+ "libc",
330
+ "pyo3-build-config",
331
+ ]
332
+
333
+ [[package]]
334
+ name = "pyo3-macros"
335
+ version = "0.22.6"
336
+ source = "registry+https://github.com/rust-lang/crates.io-index"
337
+ checksum = "0fd24d897903a9e6d80b968368a34e1525aeb719d568dba8b3d4bfa5dc67d453"
338
+ dependencies = [
339
+ "proc-macro2",
340
+ "pyo3-macros-backend",
341
+ "quote",
342
+ "syn",
343
+ ]
344
+
345
+ [[package]]
346
+ name = "pyo3-macros-backend"
347
+ version = "0.22.6"
348
+ source = "registry+https://github.com/rust-lang/crates.io-index"
349
+ checksum = "36c011a03ba1e50152b4b394b479826cad97e7a21eb52df179cd91ac411cbfbe"
350
+ dependencies = [
351
+ "heck",
352
+ "proc-macro2",
353
+ "pyo3-build-config",
354
+ "quote",
355
+ "syn",
356
+ ]
357
+
358
+ [[package]]
359
+ name = "quote"
360
+ version = "1.0.40"
361
+ source = "registry+https://github.com/rust-lang/crates.io-index"
362
+ checksum = "1885c039570dc00dcb4ff087a89e185fd56bae234ddc7f056a945bf36467248d"
363
+ dependencies = [
364
+ "proc-macro2",
365
+ ]
366
+
367
+ [[package]]
368
+ name = "rand"
369
+ version = "0.8.5"
370
+ source = "registry+https://github.com/rust-lang/crates.io-index"
371
+ checksum = "34af8d1a0e25924bc5b7c43c079c942339d8f0a8b57c39049bef581b46327404"
372
+ dependencies = [
373
+ "libc",
374
+ "rand_chacha",
375
+ "rand_core",
376
+ ]
377
+
378
+ [[package]]
379
+ name = "rand_chacha"
380
+ version = "0.3.1"
381
+ source = "registry+https://github.com/rust-lang/crates.io-index"
382
+ checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88"
383
+ dependencies = [
384
+ "ppv-lite86",
385
+ "rand_core",
386
+ ]
387
+
388
+ [[package]]
389
+ name = "rand_core"
390
+ version = "0.6.4"
391
+ source = "registry+https://github.com/rust-lang/crates.io-index"
392
+ checksum = "ec0be4795e2f6a28069bec0b5ff3e2ac9bafc99e6a9a7dc3547996c5c816922c"
393
+ dependencies = [
394
+ "getrandom",
395
+ ]
396
+
397
+ [[package]]
398
+ name = "rawpointer"
399
+ version = "0.2.1"
400
+ source = "registry+https://github.com/rust-lang/crates.io-index"
401
+ checksum = "60a357793950651c4ed0f3f52338f53b2f809f32d83a07f72909fa13e4c6c1e3"
402
+
403
+ [[package]]
404
+ name = "rayon"
405
+ version = "1.11.0"
406
+ source = "registry+https://github.com/rust-lang/crates.io-index"
407
+ checksum = "368f01d005bf8fd9b1206fb6fa653e6c4a81ceb1466406b81792d87c5677a58f"
408
+ dependencies = [
409
+ "either",
410
+ "rayon-core",
411
+ ]
412
+
413
+ [[package]]
414
+ name = "rayon-core"
415
+ version = "1.13.0"
416
+ source = "registry+https://github.com/rust-lang/crates.io-index"
417
+ checksum = "22e18b0f0062d30d4230b2e85ff77fdfe4326feb054b9783a3460d8435c8ab91"
418
+ dependencies = [
419
+ "crossbeam-deque",
420
+ "crossbeam-utils",
421
+ ]
422
+
423
+ [[package]]
424
+ name = "rustc-hash"
425
+ version = "1.1.0"
426
+ source = "registry+https://github.com/rust-lang/crates.io-index"
427
+ checksum = "08d43f7aa6b08d49f382cde6a7982047c3426db949b1424bc4b7ec9ae12c6ce2"
428
+
429
+ [[package]]
430
+ name = "rustversion"
431
+ version = "1.0.22"
432
+ source = "registry+https://github.com/rust-lang/crates.io-index"
433
+ checksum = "b39cdef0fa800fc44525c84ccb54a029961a8215f9619753635a9c0d2538d46d"
434
+
435
+ [[package]]
436
+ name = "ryu"
437
+ version = "1.0.20"
438
+ source = "registry+https://github.com/rust-lang/crates.io-index"
439
+ checksum = "28d3b2b1366ec20994f1fd18c3c594f05c5dd4bc44d8bb0c1c632c8d6829481f"
440
+
441
+ [[package]]
442
+ name = "serde"
443
+ version = "1.0.227"
444
+ source = "registry+https://github.com/rust-lang/crates.io-index"
445
+ checksum = "80ece43fc6fbed4eb5392ab50c07334d3e577cbf40997ee896fe7af40bba4245"
446
+ dependencies = [
447
+ "serde_core",
448
+ ]
449
+
450
+ [[package]]
451
+ name = "serde_core"
452
+ version = "1.0.227"
453
+ source = "registry+https://github.com/rust-lang/crates.io-index"
454
+ checksum = "7a576275b607a2c86ea29e410193df32bc680303c82f31e275bbfcafe8b33be5"
455
+ dependencies = [
456
+ "serde_derive",
457
+ ]
458
+
459
+ [[package]]
460
+ name = "serde_derive"
461
+ version = "1.0.227"
462
+ source = "registry+https://github.com/rust-lang/crates.io-index"
463
+ checksum = "51e694923b8824cf0e9b382adf0f60d4e05f348f357b38833a3fa5ed7c2ede04"
464
+ dependencies = [
465
+ "proc-macro2",
466
+ "quote",
467
+ "syn",
468
+ ]
469
+
470
+ [[package]]
471
+ name = "shlex"
472
+ version = "1.3.0"
473
+ source = "registry+https://github.com/rust-lang/crates.io-index"
474
+ checksum = "0fda2ff0d084019ba4d7c6f371c95d8fd75ce3524c3cb8fb653a3023f6323e64"
475
+
476
+ [[package]]
477
+ name = "simsimd"
478
+ version = "5.9.11"
479
+ source = "registry+https://github.com/rust-lang/crates.io-index"
480
+ checksum = "9638f2829f4887c62a01958903b58fa1b740a64d5dc2bbc4a75a33827ee1bd53"
481
+ dependencies = [
482
+ "cc",
483
+ ]
484
+
485
+ [[package]]
486
+ name = "syn"
487
+ version = "2.0.106"
488
+ source = "registry+https://github.com/rust-lang/crates.io-index"
489
+ checksum = "ede7c438028d4436d71104916910f5bb611972c5cfd7f89b8300a8186e6fada6"
490
+ dependencies = [
491
+ "proc-macro2",
492
+ "quote",
493
+ "unicode-ident",
494
+ ]
495
+
496
+ [[package]]
497
+ name = "target-lexicon"
498
+ version = "0.12.16"
499
+ source = "registry+https://github.com/rust-lang/crates.io-index"
500
+ checksum = "61c41af27dd6d1e27b1b16b489db798443478cef1f06a660c96db617ba5de3b1"
501
+
502
+ [[package]]
503
+ name = "unicode-ident"
504
+ version = "1.0.19"
505
+ source = "registry+https://github.com/rust-lang/crates.io-index"
506
+ checksum = "f63a545481291138910575129486daeaf8ac54aee4387fe7906919f7830c7d9d"
507
+
508
+ [[package]]
509
+ name = "unindent"
510
+ version = "0.2.4"
511
+ source = "registry+https://github.com/rust-lang/crates.io-index"
512
+ checksum = "7264e107f553ccae879d21fbea1d6724ac785e8c3bfc762137959b5802826ef3"
513
+
514
+ [[package]]
515
+ name = "wasi"
516
+ version = "0.11.1+wasi-snapshot-preview1"
517
+ source = "registry+https://github.com/rust-lang/crates.io-index"
518
+ checksum = "ccf3ec651a847eb01de73ccad15eb7d99f80485de043efb2f370cd654f4ea44b"
519
+
520
+ [[package]]
521
+ name = "zerocopy"
522
+ version = "0.8.27"
523
+ source = "registry+https://github.com/rust-lang/crates.io-index"
524
+ checksum = "0894878a5fa3edfd6da3f88c4805f4c8558e2b996227a3d864f47fe11e38282c"
525
+ dependencies = [
526
+ "zerocopy-derive",
527
+ ]
528
+
529
+ [[package]]
530
+ name = "zerocopy-derive"
531
+ version = "0.8.27"
532
+ source = "registry+https://github.com/rust-lang/crates.io-index"
533
+ checksum = "88d2b8d9c68ad2b9e4340d7832716a4d21a22a1154777ad56ea55c51a9cf3831"
534
+ dependencies = [
535
+ "proc-macro2",
536
+ "quote",
537
+ "syn",
538
+ ]
pkboost-0.1.0/Cargo.toml ADDED
@@ -0,0 +1,46 @@
1
+ [package]
2
+ name = "pkboost"
3
+ version = "0.1.0"
4
+ edition = "2021"
5
+ readme = "README.md"
6
+
7
+ [dependencies]
8
+ rand = "0.8"
9
+ rayon = "1.8"
10
+ csv = "1.3"
11
+ mimalloc = "0.1"
12
+ num_cpus = "1.16"
13
+ ndarray = { version = "0.16.1", features = ["rayon"] }
14
+ simsimd = "5.9"
15
+ pyo3 = { version = "0.22", features = ["extension-module"] }
16
+ numpy = "0.22"
17
+
18
+ [[bin]]
19
+ name = "benchmark"
20
+ path = "src/bin/benchmark.rs"
21
+
22
+ [[bin]]
23
+ name = "test_living"
24
+ path = "src/bin/test_living.rs"
25
+
26
+ [[bin]]
27
+ name = "test_drift"
28
+ path = "src/bin/test_drift.rs"
29
+
30
+ [[bin]]
31
+ name = "test_static"
32
+ path = "src/bin/test_static.rs"
33
+
34
+ [[bin]]
35
+ name = "test_retrain"
36
+ path = "src/bin/test_retrain.rs"
37
+
38
+ [lib]
39
+ name = "pkboost"
40
+ path = "src/lib.rs"
41
+ crate-type = ["cdylib", "rlib"]
42
+
43
+ [profile.release]
44
+ opt-level = 3
45
+ lto = "fat"
46
+ codegen-units = 1