Merge pull request #11 from CyberAgentAILab/feat/get-started-doc

Update get-started guide

Author: TomeHirata

CONTRIBUTING.md

@@ -0,0 +1,56 @@
+
+# Contribution
+Thank you for considering contributing to this project! Here are some guidelines to help you get started.
+
+## How can I contribute to this project?
+### Reporting Bugs
+If you find a bug, please report it by opening an issue in the issue tracker. Provide as much detail as possible to help us understand and reproduce the issue:
+- A clear and descriptive title.
+- A detailed description of the problem.
+- Steps to reproduce the issue.
+- Any error messages or screenshots.
+
+### Suggesting Enhancements
+We welcome suggestions for improvements! To suggest an enhancement:
+- Check the issue tracker to see if someone else has already suggested it.
+- If not, open a new issue and describe your idea clearly.
+- Explain why you believe the enhancement would be beneficial.
+
+### Pull Requests
+Pull requests are welcome! If you plan to make significant changes, please open an issue first to discuss your idea. This helps us ensure that your contribution fits with the project's direction. Follow these steps for a smooth pull request process:
+
+- Fork the repository.
+- Clone your fork to your local machine.
+- Create a new branch: `git checkout -b my-feature-branch`.
+- Make your changes.
+- Commit your changes: `git commit -m 'Add some feature'.
+- Push to the branch: `git push origin my-feature-branch`.
+- Open a pull request in the original repository.
+
+## Development
+Here are the basic commands you can use to develop this package.
+
+### Install Pipenv
+If you don't have `pipenv` installed, you can install it using `pip`:
+
+```sh
+pip install pipenv
+```
+
+### Linting
+We use `ruff` for linting the code. To run the linter, use the following command:
+```sh
+pipenv run lint
+```
+
+### Auto format
+We use `ruff` for formatting the code. To run the formatter, use the following command:
+```sh
+pipenv run format
+```
+
+### Unit test
+We use `unittest` for testing the code. To run the unit tests, use the following command:
+```sh
+pipenv run unittest
+```

README.md

@@ -1,6 +1,6 @@
 ## Overview
 
-This an a Python package for building the regression adjusted distribution function estimator proposed in "Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction".
+This a Python package for building the regression adjusted distribution function estimator proposed in "Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction". For the details of this package, see [the documentation](https://cyberagentailab.github.io/python-dte-adjustment/).
 
 ## Installation
 
@@ -17,29 +17,15 @@ This an a Python package for building the regression adjusted distribution funct
     pip install -e .
     ```
 
+## Basic Usage
+Examples of how to use this package are available in [this Get-started Guide](https://cyberagentailab.github.io/python-dte-adjustment/get_started.html).
+
 ## Development
+We welcome contributions to the project! Please review our [Contribution Guide](CONTRIBUTING.md) for details on how to get started.
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
 
-### Install Pipenv
-If you don't have `pipenv` installed, you can install it using `pip`:
-
-```sh
-pip install pipenv
-```
-
-### Linting
-We use `ruff` for linting the code. To run the linter, use the following command:
-```sh
-pipenv run lint
-```
-
-### Auto format
-We use `ruff` for formatting the code. To run the formatter, use the following command:
-```sh
-pipenv run format
-```
-
-### Unit test
-We use `unittest` for testing the code. To run the unit tests, use the following command:
-```sh
-pipenv run unittest
-```
+## Maintainers
+- [Tomu Hirata](https://github.com/TomeHirata)

docs/source/_static/dte_empirical.png

@@ -0,0 +1,4 @@
+Contribution Guide
+==================
+
+Regarding how to contribute to this package, please refer to https://github.com/CyberAgentAILab/python-dte-adjustment/CONTRIBUTING.md for more details.

docs/source/_static/dte_moment.png

@@ -43,11 +43,11 @@ Generate data for training cumulative distribution function:
       quadratic_term = np.dot(X**2, gamma)
       
       # Outcome equation
-      Y = D + linear_term + quadratic_term + U
+      Y = 5 * D + linear_term + quadratic_term + U
       
       return X, D, Y
 
-  n = 100  # Sample size
+  n = 1000  # Sample size
   X, D, Y = generate_data(n)
 
 Then, let's build an empirical cumulative distribution function (CDF).
@@ -63,13 +63,14 @@ Distributional treatment effect (DTE) can be computed easily in the following co
 
 .. code-block:: python
 
-  dte, lower_bound, upper_bound = estimator.predict_dte(target_treatment_arm=1, control_treatment_arm=0, locations=np.sort(Y), variance_type="simple")
+  locations = np.linspace(Y.min(), Y.max(), 20)
+  dte, lower_bound, upper_bound = estimator.predict_dte(target_treatment_arm=1, control_treatment_arm=0, locations=locations, variance_type="simple")
 
 A convenience function is available to visualize distribution effects. This method can be used for other distribution parameters including Probability Treatment Effect (PTE) and Quantile Treatment Effect (QTE).
 
 .. code-block:: python
 
-  plot(np.sort(Y), dte, lower_bound, upper_bound, title="DTE of simple estimator")
+  plot(locations, dte, lower_bound, upper_bound, title="DTE of simple estimator")
 
 .. image:: _static/dte_empirical.png
    :alt: DTE of empirical estimator
@@ -92,8 +93,8 @@ DTE can be computed and visualized in the following code.
 
 .. code-block:: python
 
-  dte, lower_bound, upper_bound = estimator.predict_dte(target_treatment_arm=1, control_treatment_arm=0, locations=np.sort(Y), variance_type="simple")
-  plot(np.sort(Y), dte, lower_bound, upper_bound, title="DTE of adjusted estimator with simple confidence band")
+  dte, lower_bound, upper_bound = estimator.predict_dte(target_treatment_arm=1, control_treatment_arm=0, locations=locations, variance_type="simple")
+  plot(locations, dte, lower_bound, upper_bound, title="DTE of adjusted estimator with simple confidence band")
 
 .. image:: _static/dte_simple.png
    :alt: DTE of adjusted estimator with simple confidence band
@@ -105,8 +106,8 @@ Confidence bands can be computed in different ways. In the following code, we us
 
 .. code-block:: python
 
-  dte, lower_bound, upper_bound = estimator.predict_dte(target_treatment_arm=1, control_treatment_arm=0, locations=np.sort(Y), variance_type="moment")
-  plot(np.sort(Y), dte, lower_bound, upper_bound, title="DTE of adjusted estimator with moment confidence band")
+  dte, lower_bound, upper_bound = estimator.predict_dte(target_treatment_arm=1, control_treatment_arm=0, locations=locations, variance_type="moment")
+  plot(locations, dte, lower_bound, upper_bound, title="DTE of adjusted estimator with moment confidence band")
 
 .. image:: _static/dte_moment.png
    :alt: DTE of adjusted estimator with moment confidence band
@@ -118,8 +119,8 @@ Also, an uniform confidence band is used when "uniform" is specified for the "va
 
 .. code-block:: python
 
-  dte, lower_bound, upper_bound = estimator.predict_dte(target_treatment_arm=1, control_treatment_arm=0, locations=np.sort(Y), variance_type="uniform")
-  plot(np.sort(Y), dte, lower_bound, upper_bound, title="DTE of adjusted estimator with uniform confidence band")
+  dte, lower_bound, upper_bound = estimator.predict_dte(target_treatment_arm=1, control_treatment_arm=0, locations=locations, variance_type="uniform")
+  plot(locations, dte, lower_bound, upper_bound, title="DTE of adjusted estimator with uniform confidence band")
 
 .. image:: _static/dte_uniform.png
    :alt: DTE of adjusted estimator with uniform confidence band
@@ -131,7 +132,6 @@ To compute PTE, we can use "predict_pte" method.
 
 .. code-block:: python
 
-  locations = np.linspace(Y.min(), Y.max(), 20)
   pte, lower_bound, upper_bound = estimator.predict_pte(target_treatment_arm=1, control_treatment_arm=0, width=1, locations=locations, variance_type="simple")
   plot(locations, pte, lower_bound, upper_bound, chart_type="bar", title="PTE of adjusted estimator with simple confidence band")