Skip to content

Commit b10ff87

Browse files
committed
✨ Fix docsting in TwoModels
1 parent 3a268bd commit b10ff87

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

sklift/models/models.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -222,10 +222,11 @@ class TwoModels(BaseEstimator):
222222
Args:
223223
estimator_trmnt (estimator object implementing 'fit'): The object to use to fit the treatment data.
224224
estimator_ctrl (estimator object implementing 'fit'): The object to use to fit the control data.
225-
method (string, ‘vanilla’, ’ddr_control’ or ‘ddr_treatment’, default='vanilla'): Specifies the approach:
226-
* ‘vanilla’ - two independent models
227-
* ’ddr_control’ - dependent data representation (First train control estimator)
228-
* ’ddr_treatment’ - dependent data representation (First train treatment estimator)
225+
method (string, 'vanilla', 'ddr_control' or 'ddr_treatment', default='vanilla'): Specifies the approach:
226+
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).
229230
230231
Attributes:
231232
trmnt_preds_ (array-like, shape (n_samples, )): Estimator predictions on samples when treatment.

0 commit comments

Comments
 (0)