Merge branch 'master' into master

Minyus · Apr 28, 2019 · 436c06a · 436c06a
2 parents 85262c2 + 33a3848
commit 436c06a
Show file tree

Hide file tree

Showing 9 changed files with 2,306 additions and 2,205 deletions.
diff --git a/.gitignore b/.gitignore
@@ -107,5 +107,13 @@ venv.bak/
 # PyCharm
 .idea/
 
+# pipenv files
+Pipfile
+Pipfile.lock
+
+# Output CSV files
+examples/CATE_for_Train.csv
+examples/CATE_for_Test.csv
+
 # vscode
-.vscode/
+.vscode/
diff --git a/causallift/__init__.py b/causallift/__init__.py
@@ -17,4 +17,4 @@
 from .estimate_propensity import estimate_propensity
 from .causal_lift import CausalLift
 
-__version__='0.0.1'
+__version__='0.0.2'
diff --git a/causallift/causal_lift.py b/causallift/causal_lift.py
@@ -1,5 +1,7 @@
 """ causal_lift.py """
 
+from IPython.display import display
+
 from .utils import (get_cols_features,
                     concat_train_test,
                     concat_train_test_df,

diff --git a/causallift/estimate_propensity.py b/causallift/estimate_propensity.py
@@ -1,5 +1,6 @@
 """ estimate_propensity.py """
 
+from IPython.display import display
 from .utils import (get_cols_features,
                     concat_train_test,
                     concat_train_test_df,

diff --git a/causallift/model_for_each.py b/causallift/model_for_each.py
@@ -1,6 +1,8 @@
 """ model_for_each.py """
 """ 2 supervised models """
 
+from IPython.display import display
+
 from .utils import (get_cols_features,
                     concat_train_test,
                     concat_train_test_df,
@@ -73,7 +75,7 @@ def __init__(self,
                 (1 / propensity) if treatment_val == 1.0 else (1 / (1 - propensity))
         else:
             # do not use sample weight
-            sample_weight = 1.0
+            sample_weight = np.ones_like(y_train, dtype=float)
 
         if params is None:
             params = {