Skip to content

Commit c740435

Browse files
author
arnaudde
committed
Rename functions and clean up
1 parent 7aeafa2 commit c740435

1 file changed

Lines changed: 8 additions & 24 deletions

File tree

dataikuapi/dss/ml.py

Lines changed: 8 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -431,15 +431,6 @@ def get_split_params(self):
431431
"""
432432
return PredictionSplitParamsHandler(self.mltask_settings)
433433

434-
@property
435-
def assertions_params(self):
436-
"""
437-
Retrieves the assertions parameters for this ml task
438-
439-
:rtype: :class:`dataikuapi.dss.ml.DSSMLAssertionsParams`
440-
"""
441-
return self.get_assertions_params()
442-
443434
def get_assertions_params(self):
444435
"""
445436
Retrieves the assertions parameters for this ml task
@@ -899,7 +890,7 @@ def expected_max(self, expected_max):
899890
class DSSMLAssertionsMetrics(object):
900891
"""
901892
Object that represents the per assertion metrics for all assertions on a trained model
902-
Do not create this object directly, use :meth:`dataikuapi.dss.ml.DSSTrainedPredictionModelDetails.get_per_assertion_metrics()` instead
893+
Do not create this object directly, use :meth:`dataikuapi.dss.ml.DSSTrainedPredictionModelDetails.get_assertions_metrics()` instead
903894
"""
904895
def __init__(self, data):
905896
self._internal_dict = data
@@ -911,13 +902,14 @@ def get_raw(self):
911902
"""
912903
return self._internal_dict
913904

914-
def get_per_assertion_metric(self, assertion_name):
905+
def get_metric(self, assertion_name):
915906
"""
916-
Retrieves the metric computed for this trained model for the assertion with the provided name (or None)
907+
Retrieves the metric computed for this trained model for the assertion with the provided name (or None if no
908+
assertion with that name exists)
917909
918910
:param str assertion_name: Name of the assertion
919911
920-
:returns: an object representing assertion metrics
912+
:returns: an object representing assertion metrics or None if if no assertion with that name exists
921913
:rtype: :class:`dataikuapi.dss.ml.DSSMLAssertionMetric`
922914
"""
923915
for assertion_metric_dict in self._internal_dict["perAssertion"]:
@@ -937,7 +929,7 @@ def positive_assertion_ratio(self):
937929
class DSSMLAssertionMetric(object):
938930
"""
939931
Object that represents the result of an assertion on a trained model
940-
Do not create this object directly, use :meth:`dataikuapi.dss.ml.DSSMLPerAssertionMetrics.get_assertion_metric(self, assertion_name)` instead
932+
Do not create this object directly, use :meth:`dataikuapi.dss.ml.DSSMLAssertionMetrics.get_metric(self, assertion_name)` instead
941933
"""
942934
def __init__(self, data):
943935
self._internal_dict = data
@@ -968,7 +960,8 @@ def result(self):
968960
@property
969961
def valid_ratio(self):
970962
"""
971-
Returns the ratio of passing rows in the assertion population
963+
Returns the ratio of rows in the assertion population with prediction equals to the expected class
964+
for classification or in the expected range for regression
972965
:rtype: float
973966
"""
974967
return self._internal_dict["validRatio"]
@@ -1148,15 +1141,6 @@ def get_performance_metrics(self):
11481141
del clean_snippet[x]
11491142
return clean_snippet
11501143

1151-
@property
1152-
def assertions_metrics(self):
1153-
"""
1154-
Retrieves assertions metrics computed for this trained model
1155-
:returns: an object representing assertion metrics
1156-
:rtype: :class:`dataikuapi.dss.ml.DSSMLAssertionsMetrics`
1157-
"""
1158-
return self.get_assertions_metrics()
1159-
11601144
def get_assertions_metrics(self):
11611145
"""
11621146
Retrieves assertions metrics computed for this trained model

0 commit comments

Comments
 (0)