Commit Graph

23 Commits

Author SHA1 Message Date
Tialo 5ced13cd6e
ENH Add Array API compatibility for `entropy` (#29141) 2024-06-14 07:29:18 +00:00
Olivier Grisel 78102fdb75
FIX always return None as device when array API dispatch is disabled (#29119) 2024-05-28 10:41:24 +05:00
Omar Salman acd2d90e50
ENH Array API support for LabelEncoder (#27381)
Co-authored-by: Omar Salman <omar.salman@arbisoft>
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
2024-05-16 15:47:35 +02:00
Thomas Li e12f192c58
ENH Use Array API in mean_tweedie_deviance (#28106) 2024-05-08 11:21:27 +05:00
Franck Charras 171e124656
FEA: Ridge support for Array API compliant inputs (#27800)
Co-authored-by: Eric Lindgren <ericlin@chalmers.se>
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
Co-authored-by: Tim Head <betatim@gmail.com>
Co-authored-by: Loïc Estève <loic.esteve@ymail.com>
Co-authored-by: Oleksii Kachaiev <kachayev@gmail.com>
Co-authored-by: Julien Jerphanion <git@jjerphan.xyz>
Co-authored-by: Guillaume Lemaitre <g.lemaitre58@gmail.com>
2024-04-25 11:52:04 +02:00
Ralf Gommers 5efc66740c
MAINT: convert `numpy.array_api` to array-api-strict (#28555)
Co-authored-by: adrinjalali <adrin.jalali@gmail.com>
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
Co-authored-by: Tim Head <betatim@gmail.com>
2024-03-18 17:27:09 +00:00
Tim Head e5a7c3e39f
FIX Fix array API `train_test_split` (#28407)
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
2024-03-18 15:43:09 +01:00
Franck Charras 612d93da5e
ENH Use Array API in `r2_score` (#27904)
Signed-off-by: Julien Jerphanion <git@jjerphan.xyz>
Co-authored-by: Eric Lindgren <ericlin@chalmers.se>
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
Co-authored-by: Tim Head <betatim@gmail.com>
Co-authored-by: Loïc Estève <loic.esteve@ymail.com>
Co-authored-by: Oleksii Kachaiev <kachayev@gmail.com>
Co-authored-by: Julien Jerphanion <git@jjerphan.xyz>
Co-authored-by: Guillaume Lemaitre <g.lemaitre58@gmail.com>
2024-03-11 16:00:46 +00:00
Franck Charras 3b212bdbc3
Small cleaning of `_atol_for_dtype` and `get_namespace` usage for consistency (#28057)
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
2024-01-09 10:52:25 +00:00
Franck Charras 77aeb825b6
FIX Apply dtype param in `check_array_api_compute_metric` unit test (#27940) 2023-12-18 16:43:08 +01:00
Franck Charras 8e10cd7f0c
MNT remove `take` fn in array_api wrapper (#27939) 2023-12-11 16:41:29 -05:00
Edoardo Abati 844b087da4
ENH Add Array API compatibility to `zero_one_loss` and `accuracy_score` (#27137) 2023-09-07 16:03:15 +02:00
Edoardo Abati 2efe3f3f6b
ENH Add Array API compatibility to MaxAbsScaler (#27110) 2023-08-23 16:09:28 +02:00
Tim Head 061f8777b4
ENH Add Array API compatibility to MinMaxScaler (#26243)
Co-authored-by: Thomas J. Fan <thomasjpfan@gmail.com>
2023-08-18 14:39:08 +00:00
Edoardo Abati ccf7d53058
Add `mps` device in Array API tests (#27098)
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
2023-08-18 14:33:33 +00:00
Tim Head 1b0a51bd90
Add tests for train_test_split with Array API input (#26855)
Co-authored-by: Thomas J. Fan <thomasjpfan@gmail.com>
2023-08-09 18:27:31 +00:00
Mateusz Sokół 702316c271
ENH Array API support for PCA (#26315)
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
Co-authored-by: Tim Head <betatim@gmail.com>
Co-authored-by: Thomas J. Fan <thomasjpfan@gmail.com>
2023-07-13 16:58:06 -05:00
Adrin Jalali 42173fdb34
MNT add isort to ruff's rules (#26649) 2023-06-21 17:50:07 +02:00
Thomas J. Fan 66a3905541
ENH Adds PyTorch support to LinearDiscriminantAnalysis (#25956)
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
Co-authored-by: Tim Head <betatim@gmail.com>
2023-04-21 11:13:29 +02:00
Thomas J. Fan 87ec7401cc
ENH Forces shape to be tuple when using Array API's reshape (#26030)
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
Co-authored-by: Tim Head <betatim@gmail.com>
2023-04-04 11:03:00 +02:00
Thomas J. Fan f097f43645
ENH Adds isdtype to Array API wrapper (#26029) 2023-04-03 14:31:09 +02:00
Thomas J. Fan bdc5df95f7
CLN Make _NumPyAPIWrapper naming consistent to _ArrayAPIWrapper (#26039) 2023-04-02 09:04:39 -04:00
Thomas J. Fan 2710a9e7ee
ENH Adds Array API support to LinearDiscriminantAnalysis (#22554)
Co-authored-by: Olivier Grisel <olivier.grisel@ensta.org>
Co-authored-by: Julien Jerphanion <git@jjerphan.xyz>
2022-09-21 15:36:19 +02:00