make AutoML.classes_ an array (#467)

* remove .tolist()

* docstr
This commit is contained in:
Chi Wang 2022-02-25 22:13:41 -08:00 committed by GitHub
parent 6095be039f
commit e3e737c71a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions

View File

@ -711,13 +711,13 @@ class AutoML(BaseEstimator):
@property
def classes_(self):
"""A list of n_classes elements for class labels."""
"""A numpy array of shape (n_classes,) for class labels."""
attr = getattr(self, "_label_transformer", None)
if attr:
return attr.classes_.tolist()
return attr.classes_
attr = getattr(self, "_trained_estimator", None)
if attr:
return attr.classes_.tolist()
return attr.classes_
return None
@property

View File

@ -1 +1 @@
__version__ = "0.9.7"
__version__ = "0.10.0"