amaiya / causalnlp Goto Github PK

View Code? Open in Web Editor NEW

130.0 6.0 11.0 15.34 MB

CausalNLP is a practical toolkit for causal inference with text as treatment, outcome, or "controlled-for" variable.

Home Page: https://amaiya.github.io/causalnlp/

License: Apache License 2.0

Makefile 0.05% Python 25.41% Jupyter Notebook 74.53%

causal-inference nlp

causalnlp's Issues

Does your model support other languages than English?

Hi Amaiya,
Thanks for your great package.
Would you kindly let me know if your package supports languages other than English when using CausalBert?

I'm also interested in knowing whether I can exploit other Transformers models from the Huggingface hub?

Error while fitting the model

Hi,

I ran to this bug while fitting the model. I checked the data and everything looks good. I don't get the root cause of this error.

File /opt/conda/lib/python3.8/site-packages/causalnlp/meta/slearner.py:80, in BaseSLearner.fit(self, X, treatment, y, p)
     78 mask = (treatment == group) | (treatment == self.control_name)
     79 treatment_filt = treatment[mask]
---> 80 X_filt = X[mask]
     81 y_filt = y[mask]
     83 w = (treatment_filt == group).astype(int)

IndexError: boolean index did not match indexed array along dimension 0

Recommend Projects