Skip to content

Commit 453d44c

Browse files
authored
Update README.md (#164)
1 parent 3e97433 commit 453d44c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ head(predict(opt.tree, X[-train, ]))
7474
`policy_tree()`: fits a depth _k_ tree by exhaustive search (_Nxp_ features on _Nxd_ actions). The optimal tree maximizes the sum of rewards: let $\Gamma_i \in \mathbb R^d$ be a vector of unit-specific rewards for each action 1 to $d$ and $\pi(X_i) \in \\{1, ..., d\\}$ a mapping from covariates $X_i$ to action. `policy_tree` solves the following:
7575

7676
$$
77-
\pi^* = argmax_{\pi \in \Pi} \left[ \sum_{i=1}^{n} \Gamma_i(\pi(X_i)) \right],
77+
\pi^* = argmax_{\pi \in \Pi} \left[\frac{1}{n} \sum_{i=1}^{n} \Gamma_i(\pi(X_i)) \right],
7878
$$
7979

8080
where $\Pi$ is the class of depth-_k_ decision trees. (`hybrid_policy_tree()` employs a mix between a optimal/greedy approach and can be used to fit deeper trees).

0 commit comments

Comments
 (0)