Skip to content

Commit e5973c6

Browse files
committed
enable log of precision recall graphs
1 parent f3d35f4 commit e5973c6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/smexperiments/tracker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -450,7 +450,7 @@ def log_table(self, title=None, values=None, data_frame=None, output_artifact=Tr
450450

451451
self._log_graph_artifact(title, data, "Table", output_artifact)
452452

453-
def _log_precision_recall(
453+
def log_precision_recall(
454454
self,
455455
y_true,
456456
predicted_probabilities,

tests/unit/test_tracker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ def test_log_pr_curve(under_test):
349349

350350
under_test._artifact_uploader.upload_object_artifact.return_value = ("s3uri_value", "etag_value")
351351

352-
under_test._log_precision_recall(y_true, y_scores, title="TestPRCurve")
352+
under_test.log_precision_recall(y_true, y_scores, title="TestPRCurve")
353353

354354
expected_data = {
355355
"type": "PrecisionRecallCurve",

0 commit comments

Comments
 (0)