Skip to content

Commit 4461201

Browse files
committed
Update double_ml.py
1 parent 72fe6b8 commit 4461201

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

doubleml/double_ml.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1049,7 +1049,8 @@ def evaluate_learners(self, learners=None, metric=mean_squared_error):
10491049
A list of strings which correspond to the nuisance functions of the model.
10501050
10511051
metric : callable
1052-
A callable function with inputs ``y_pred`` and ``y_true``.
1052+
A callable function with inputs ``y_pred`` and ``y_true`` of shape ``(1, n)``,
1053+
where ``n`` specifies the number of observations.
10531054
Default is the euclidean distance.
10541055
10551056
Returns

0 commit comments

Comments
 (0)