mirror of
https://github.com/microsoft/autogen.git
synced 2025-11-17 18:44:47 +00:00
Expose feature and label transformer in automl.py (#993)
* expose label and feature transformer * linter apply * avoid undefined attribute in flaml/automl/automl.py Co-authored-by: Chi Wang <wang.chi@microsoft.com> * avoid undefined attribute in flaml/automl/automl.py Co-authored-by: Chi Wang <wang.chi@microsoft.com> * retrigger checks * retrigger checks --------- Co-authored-by: Chi Wang <wang.chi@microsoft.com>
This commit is contained in:
parent
c9fc622af1
commit
b235fe0098
@ -490,6 +490,16 @@ class AutoML(BaseEstimator):
|
|||||||
with open(filename, "w") as f:
|
with open(filename, "w") as f:
|
||||||
json.dump(best, f)
|
json.dump(best, f)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def feature_transformer(self):
|
||||||
|
"""Returns AutoML Transformer"""
|
||||||
|
return getattr(self, "_transformer", None)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def label_transformer(self):
|
||||||
|
"""Returns AutoML label transformer"""
|
||||||
|
return getattr(self, "_label_transformer", None)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def classes_(self):
|
def classes_(self):
|
||||||
"""A numpy array of shape (n_classes,) for class labels."""
|
"""A numpy array of shape (n_classes,) for class labels."""
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user