File tree 1 file changed +5
-1
lines changed
google/cloud/aiplatform/metadata
1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -1009,7 +1009,7 @@ def log_classification_metrics(
1009
1009
tpr : Optional [List [float ]] = None ,
1010
1010
threshold : Optional [List [float ]] = None ,
1011
1011
display_name : Optional [str ] = None ,
1012
- ):
1012
+ ) -> google_artifact_schema . ClassificationMetrics :
1013
1013
"""Create an artifact for classification metrics and log to ExperimentRun. Currently supports confusion matrix and ROC curve.
1014
1014
1015
1015
```
@@ -1038,6 +1038,9 @@ def log_classification_metrics(
1038
1038
display_name (str):
1039
1039
Optional. The user-defined name for the classification metric artifact.
1040
1040
1041
+ Returns:
1042
+ ClassificationMetrics artifact.
1043
+
1041
1044
Raises:
1042
1045
ValueError: if 'labels' and 'matrix' are not set together
1043
1046
or if 'labels' and 'matrix' are not in the same length
@@ -1102,6 +1105,7 @@ def log_classification_metrics(
1102
1105
self ._metadata_node .add_artifacts_and_executions (
1103
1106
artifact_resource_names = [classfication_metrics .resource_name ]
1104
1107
)
1108
+ return classification_metrics
1105
1109
1106
1110
@_v1_not_supported
1107
1111
def log_model (
You can’t perform that action at this time.
0 commit comments