autogluon.core 1.2.1b20250415__py3-none-any.whl → 1.2.1b20250417__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.
- autogluon/core/hpo/executors.py +6 -3
- autogluon/core/metrics/classification_metrics.py +8 -1
- autogluon/core/version.py +1 -1
- {autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/METADATA +7 -7
- {autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/RECORD +12 -12
- /autogluon.core-1.2.1b20250415-py3.9-nspkg.pth → /autogluon.core-1.2.1b20250417-py3.9-nspkg.pth +0 -0
- {autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/LICENSE +0 -0
- {autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/NOTICE +0 -0
- {autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/WHEEL +0 -0
- {autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/namespace_packages.txt +0 -0
- {autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/top_level.txt +0 -0
- {autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/zip-safe +0 -0
autogluon/core/hpo/executors.py
CHANGED
@@ -6,7 +6,7 @@ import math
|
|
6
6
|
import os
|
7
7
|
import time
|
8
8
|
from abc import ABC, abstractmethod
|
9
|
-
from typing import TYPE_CHECKING, Any, Callable, Dict,
|
9
|
+
from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple, Union
|
10
10
|
|
11
11
|
import pandas as pd
|
12
12
|
|
@@ -14,7 +14,6 @@ from autogluon.common import space
|
|
14
14
|
from autogluon.common.utils.resource_utils import ResourceManager
|
15
15
|
from autogluon.common.utils.s3_utils import is_s3_url
|
16
16
|
|
17
|
-
from ..ray.resources_calculator import ResourceCalculator
|
18
17
|
from ..scheduler.scheduler_factory import scheduler_factory
|
19
18
|
from ..utils.savers import save_pkl
|
20
19
|
from .constants import CUSTOM_BACKEND, RAY_BACKEND
|
@@ -135,7 +134,11 @@ class HpoExecutor(ABC):
|
|
135
134
|
resources_per_trial = self.hyperparameter_tune_kwargs["resources_per_trial"]
|
136
135
|
total_num_cpus_per_trial = resources_per_trial.get("num_cpus")
|
137
136
|
total_num_gpus_per_trial = resources_per_trial.get("num_gpus")
|
138
|
-
|
137
|
+
|
138
|
+
if hasattr(model_base, "_user_params_aux"):
|
139
|
+
user_specified_fold_resources = model_base._user_params_aux
|
140
|
+
else:
|
141
|
+
user_specified_fold_resources = {}
|
139
142
|
user_specified_fold_num_cpus = user_specified_fold_resources.get("num_cpus", None) # We shouldn't always use it
|
140
143
|
user_specified_fold_num_gpus = user_specified_fold_resources.get("num_gpus", None)
|
141
144
|
if user_specified_fold_num_cpus is not None or user_specified_fold_num_gpus is not None:
|
@@ -14,6 +14,13 @@ try:
|
|
14
14
|
except:
|
15
15
|
from sklearn.metrics.classification import _check_targets, type_of_target
|
16
16
|
|
17
|
+
try:
|
18
|
+
# numpy>=2
|
19
|
+
from numpy import trapezoid as trapezoid
|
20
|
+
except:
|
21
|
+
# numpy<2, deprecated in numpy>=2
|
22
|
+
from numpy import trapz as trapezoid
|
23
|
+
|
17
24
|
logger = logging.getLogger(__name__)
|
18
25
|
|
19
26
|
|
@@ -434,4 +441,4 @@ def customized_binary_roc_auc_score(y_true: Union[np.array, pd.Series], y_score:
|
|
434
441
|
raise ValueError("Only one class present in y_true. ROC AUC score is not defined in that case.")
|
435
442
|
fpr = fps / fps[-1]
|
436
443
|
tpr = tps / tps[-1]
|
437
|
-
return
|
444
|
+
return trapezoid(tpr, fpr)
|
autogluon/core/version.py
CHANGED
{autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/METADATA
RENAMED
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.1
|
2
2
|
Name: autogluon.core
|
3
|
-
Version: 1.2.
|
3
|
+
Version: 1.2.1b20250417
|
4
4
|
Summary: Fast and Accurate ML in 3 Lines of Code
|
5
5
|
Home-page: https://github.com/autogluon/autogluon
|
6
6
|
Author: AutoGluon Community
|
@@ -45,12 +45,12 @@ Requires-Dist: tqdm<5,>=4.38
|
|
45
45
|
Requires-Dist: requests
|
46
46
|
Requires-Dist: matplotlib<3.11,>=3.7.0
|
47
47
|
Requires-Dist: boto3<2,>=1.10
|
48
|
-
Requires-Dist: autogluon.common==1.2.
|
48
|
+
Requires-Dist: autogluon.common==1.2.1b20250417
|
49
49
|
Provides-Extra: all
|
50
|
-
Requires-Dist:
|
50
|
+
Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
|
51
51
|
Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "all"
|
52
|
+
Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "all"
|
52
53
|
Requires-Dist: pyarrow>=15.0.0; extra == "all"
|
53
|
-
Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
|
54
54
|
Provides-Extra: ray
|
55
55
|
Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "ray"
|
56
56
|
Provides-Extra: raytune
|
@@ -58,11 +58,11 @@ Requires-Dist: pyarrow>=15.0.0; extra == "raytune"
|
|
58
58
|
Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "raytune"
|
59
59
|
Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "raytune"
|
60
60
|
Provides-Extra: tests
|
61
|
-
Requires-Dist: flake8; extra == "tests"
|
62
|
-
Requires-Dist: pytest-mypy; extra == "tests"
|
63
|
-
Requires-Dist: types-setuptools; extra == "tests"
|
64
61
|
Requires-Dist: pytest; extra == "tests"
|
62
|
+
Requires-Dist: types-setuptools; extra == "tests"
|
65
63
|
Requires-Dist: types-requests; extra == "tests"
|
64
|
+
Requires-Dist: flake8; extra == "tests"
|
65
|
+
Requires-Dist: pytest-mypy; extra == "tests"
|
66
66
|
|
67
67
|
|
68
68
|
|
@@ -1,9 +1,9 @@
|
|
1
|
-
autogluon.core-1.2.
|
1
|
+
autogluon.core-1.2.1b20250417-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
|
2
2
|
autogluon/core/__init__.py,sha256=8KfvvHzXX3a4q6z43Dw1yE7VtbAoiSMaglVpKDy6Xeg,245
|
3
3
|
autogluon/core/_setup_utils.py,sha256=TnDboM66MzSp0CLACo_vcNNwhORQWE8JL0kt1IYa7dI,7014
|
4
4
|
autogluon/core/constants.py,sha256=nEVLdSFJ-5O-tz3jUD3qPX65RMp7g8qOR38XlurbP4Y,3403
|
5
5
|
autogluon/core/problem_type.py,sha256=XJmMgeNBgS7u43pDK-spTivatPyh_INOXveEXwQt-Rw,2993
|
6
|
-
autogluon/core/version.py,sha256=
|
6
|
+
autogluon/core/version.py,sha256=9DNociU3CkjmMks3rFL-2UfsFfw26T7_cq6VAom-ic4,91
|
7
7
|
autogluon/core/augmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
8
8
|
autogluon/core/augmentation/distill_utils.py,sha256=JBlp2WOMNKoJv8aKVwJVRQSalSk8jx36HM7-k_VvkhY,9404
|
9
9
|
autogluon/core/calibrate/__init__.py,sha256=eU6qLj7DKUhaz2HHNHDrfroRaLM-mhuSncK_v1UP4F8,62
|
@@ -21,7 +21,7 @@ autogluon/core/data/label_cleaner.py,sha256=rgXFICs3oNlkaglnSohwjh-vcSaN2mcJTDAJ
|
|
21
21
|
autogluon/core/hpo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
22
22
|
autogluon/core/hpo/constants.py,sha256=vYeJNySZtldqKX3hctMM-RMT8UuVZneH3tAneiIaIZM,117
|
23
23
|
autogluon/core/hpo/exceptions.py,sha256=TvkCQZSd-TMwIqmxkOAONMuiz-HimxwHwjvGynOj5ug,44
|
24
|
-
autogluon/core/hpo/executors.py,sha256=
|
24
|
+
autogluon/core/hpo/executors.py,sha256=uXt-xR9imDBx9Uf4E03QcLZqYfzrnyPPvj5yuhLQuYg,30335
|
25
25
|
autogluon/core/hpo/ray_hpo.py,sha256=6Dn5SRo7xeeB1B-B7bryOAgkO7yh_Z2QLDx5enNkk9U,21523
|
26
26
|
autogluon/core/hpo/ray_tune_constants.py,sha256=Lfx12RIRhSl9WgVpCQqEEtwWtvBIr6SCXNmeYuKZ88k,220
|
27
27
|
autogluon/core/hpo/ray_tune_scheduler.py,sha256=T9xwB3FjTTnAEOhiqkB4eev21lQKU6K4hsx_WsS9-Lg,1112
|
@@ -32,7 +32,7 @@ autogluon/core/learner/__init__.py,sha256=pAEQBcY-xq-HG2QbHdt9V4iq1RSlz6gfcxNa40
|
|
32
32
|
autogluon/core/learner/abstract_learner.py,sha256=A4nAR-U27wInLtghQjpLdEhBgh9M_2DFz7NhoVhjuL4,5784
|
33
33
|
autogluon/core/learning_curves/plot_curves.py,sha256=GxWeX6gvp9XlA2QwtBi9cdaxjtbY3mSngWXl-EjM8bE,3141
|
34
34
|
autogluon/core/metrics/__init__.py,sha256=7UTGdSwuRG4Wjfenyvj9RBYMtTSGAyPpy0Q_ktF-AF8,29827
|
35
|
-
autogluon/core/metrics/classification_metrics.py,sha256
|
35
|
+
autogluon/core/metrics/classification_metrics.py,sha256=-1osD_IH7fOA7O-iDAJwYpY0a4ZeD451M82GnwXSxJA,18496
|
36
36
|
autogluon/core/metrics/quantile_metrics.py,sha256=yjnKSMuLdee4vemXIoxitCJWf_3lvE-RXlJ4V1371w4,1393
|
37
37
|
autogluon/core/metrics/score_func.py,sha256=RIFOZMhPKYRFXBufoe5XIY1BRDr8aqGA-2zVwuTKi0A,4684
|
38
38
|
autogluon/core/metrics/softclass_metrics.py,sha256=inn35DfftLZey0mK3OuMJPzm58eZgQ38vnEBO8sZ3mg,1272
|
@@ -89,11 +89,11 @@ autogluon/core/utils/utils.py,sha256=FMa9kIUAxA3IIBbATmBnNEVObSAivehZ2_zCy3PRR-c
|
|
89
89
|
autogluon/core/utils/version_utils.py,sha256=5-r8hLRKTaZbj5qo2uzE_2E4casH49Ye3WyeHlgHuz4,3252
|
90
90
|
autogluon/core/utils/loaders/__init__.py,sha256=W5FAdQvpDcn_uisqJrlSAObWVta-YjJLKGN3NCbEgIo,109
|
91
91
|
autogluon/core/utils/savers/__init__.py,sha256=bGWciSxAkj6u06vOC4pTvr22f_1ey0glgvmjCMEOm78,89
|
92
|
-
autogluon.core-1.2.
|
93
|
-
autogluon.core-1.2.
|
94
|
-
autogluon.core-1.2.
|
95
|
-
autogluon.core-1.2.
|
96
|
-
autogluon.core-1.2.
|
97
|
-
autogluon.core-1.2.
|
98
|
-
autogluon.core-1.2.
|
99
|
-
autogluon.core-1.2.
|
92
|
+
autogluon.core-1.2.1b20250417.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
|
93
|
+
autogluon.core-1.2.1b20250417.dist-info/METADATA,sha256=WFLhQI8Kh5SJ8AdlKK6wvNcZrFZlMqmTjsJ8gtqGe6s,12399
|
94
|
+
autogluon.core-1.2.1b20250417.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
|
95
|
+
autogluon.core-1.2.1b20250417.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
|
96
|
+
autogluon.core-1.2.1b20250417.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
|
97
|
+
autogluon.core-1.2.1b20250417.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
|
98
|
+
autogluon.core-1.2.1b20250417.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
|
99
|
+
autogluon.core-1.2.1b20250417.dist-info/RECORD,,
|
/autogluon.core-1.2.1b20250415-py3.9-nspkg.pth → /autogluon.core-1.2.1b20250417-py3.9-nspkg.pth
RENAMED
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
{autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/top_level.txt
RENAMED
File without changes
|
{autogluon.core-1.2.1b20250415.dist-info → autogluon.core-1.2.1b20250417.dist-info}/zip-safe
RENAMED
File without changes
|