Skip to content

Commit 5097b81

Browse files
committed
✨ Fix docsting in TwoModels
1 parent 4fc5ee8 commit 5097b81

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

sklift/models/models.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -224,9 +224,12 @@ class TwoModels(BaseEstimator):
224224
estimator_ctrl (estimator object implementing 'fit'): The object to use to fit the control data.
225225
method (string, 'vanilla', 'ddr_control' or 'ddr_treatment', default='vanilla'): Specifies the approach:
226226
227-
* ``'vanilla'`` - two independent models;
228-
* ``'ddr_control'`` - dependent data representation (First train control estimator);
229-
* ``'ddr_treatment'`` - dependent data representation (First train treatment estimator).
227+
* ``'vanilla'``:
228+
Two independent models;
229+
* ``'ddr_control'``:
230+
Dependent data representation (First train control estimator).
231+
* ``'ddr_treatment'``:
232+
Dependent data representation (First train treatment estimator).
230233
231234
Attributes:
232235
trmnt_preds_ (array-like, shape (n_samples, )): Estimator predictions on samples when treatment.

0 commit comments

Comments
 (0)