Skip to content

Commit 58faf71

Browse files
committed
📝 Fix y_true description in docstrings
1 parent aa9ef3d commit 58faf71

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

sklift/metrics/metrics.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def uplift_curve(y_true, uplift, treatment):
1212
area under the Uplift Curve, see :func:`uplift_auc_score`.
1313
1414
Args:
15-
y_true (1d array-like): Ground truth (correct) labels.
15+
y_true (1d array-like): Correct (true) target values.
1616
uplift (1d array-like): Predicted uplift, as returned by a model.
1717
treatment (1d array-like): Treatment labels.
1818
@@ -48,7 +48,7 @@ def uplift_curve(y_true, uplift, treatment):
4848
num_ctrl = num_all - num_trmnt
4949
y_ctrl = stable_cumsum(y_true_ctrl)[threshold_indices]
5050

51-
curve_values = (np.divide(y_trmnt, num_trmnt, out=np.zeros_like(y_trmnt), where=num_trmnt != 0) -\
51+
curve_values = (np.divide(y_trmnt, num_trmnt, out=np.zeros_like(y_trmnt), where=num_trmnt != 0) -
5252
np.divide(y_ctrl, num_ctrl, out=np.zeros_like(y_ctrl), where=num_ctrl != 0)) * num_all
5353

5454
if num_all.size == 0 or curve_values[0] != 0 or num_all[0] != 0:
@@ -67,7 +67,7 @@ def qini_curve(y_true, uplift, treatment):
6767
area under the Qini Curve, see :func:`qini_auc_score`.
6868
6969
Args:
70-
y_true (1d array-like): Ground truth (correct) labels.
70+
y_true (1d array-like): Correct (true) target values.
7171
uplift (1d array-like): Predicted uplift, as returned by a model.
7272
treatment (1d array-like): Treatment labels.
7373
@@ -120,7 +120,7 @@ def uplift_auc_score(y_true, uplift, treatment):
120120
"""Compute Area Under the Uplift Curve from prediction scores.
121121
122122
Args:
123-
y_true (1d array-like): Ground truth (correct) labels.
123+
y_true (1d array-like): Correct (true) target values.
124124
uplift (1d array-like): Predicted uplift, as returned by a model.
125125
treatment (1d array-like): Treatment labels.
126126
@@ -136,7 +136,7 @@ def auuc(y_true, uplift, treatment):
136136
"""Compute Area Under the Uplift Curve from prediction scores.
137137
138138
Args:
139-
y_true (1d array-like): Ground truth (correct) labels.
139+
y_true (1d array-like): Correct (true) target values.
140140
uplift (1d array-like): Predicted uplift, as returned by a model.
141141
treatment (1d array-like): Treatment labels.
142142
@@ -160,7 +160,7 @@ def qini_auc_score(y_true, uplift, treatment):
160160
"""Compute Area Under the Qini Curve (aka Qini coefficient) from prediction scores.
161161
162162
Args:
163-
y_true (1d array-like): Ground truth (correct) labels.
163+
y_true (1d array-like): Correct (true) target values.
164164
uplift (1d array-like): Predicted uplift, as returned by a model.
165165
treatment (1d array-like): Treatment labels.
166166
@@ -176,7 +176,7 @@ def auqc(y_true, uplift, treatment):
176176
"""Compute Area Under the Qini Curve (aka Qini coefficient) from prediction scores.
177177
178178
Args:
179-
y_true (1d array-like): Ground truth (correct) labels.
179+
y_true (1d array-like): Correct (true) target values.
180180
uplift (1d array-like): Predicted uplift, as returned by a model.
181181
treatment (1d array-like): Treatment labels.
182182
@@ -200,7 +200,7 @@ def uplift_at_k(y_true, uplift, treatment, strategy, k=0.3):
200200
"""Compute uplift at first k percentage of the total sample.
201201
202202
Args:
203-
y_true (1d array-like): Ground truth (correct) labels.
203+
y_true (1d array-like): Correct (true) target values.
204204
uplift (1d array-like): Predicted uplift, as returned by a model.
205205
treatment (1d array-like): Treatment labels.
206206
k (float or int): If float, should be between 0.0 and 1.0 and represent the proportion of the dataset

0 commit comments

Comments
 (0)