Skip to content

Commit

Permalink
fix missing imports
Browse files Browse the repository at this point in the history
- Add the import for the data generator.
- Add the import for expit (sigmoid function)
  • Loading branch information
Tanmay-Kulkarni101 authored and MaximilianFranz committed Feb 25, 2020
1 parent 33e6ea2 commit 9142a0f
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion docs/usage.rst
Original file line number Diff line number Diff line change
Expand Up @@ -438,7 +438,9 @@ where
To implement that as a DGP in JustCause we define the outcome function as follows::

from sklearn.utils import check_random_state # ensures usable random state

from justcause.data.utils import generate_data
from scipy.special import expit
def outcome(covariates, *, random_state: RandomState, **kwargs):
random_state = check_random_state(random_state)

Expand Down

0 comments on commit 9142a0f

Please sign in to comment.