Skip to content

Commit

Permalink
Minor README changes
Browse files Browse the repository at this point in the history
  • Loading branch information
andrewtavis committed Feb 3, 2020
1 parent 6c40d25 commit 31423e0
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -207,6 +207,7 @@ Quickly iterate models to derive their average effects and prediction variance.
```python
from causeinfer.evaluation import iterate_model, eval_table

n = num_iterations
avg_preds, all_preds, \
avg_eval, eval_variance, \
eval_sd, all_evals = iterate_model(model=model,
Expand All @@ -218,7 +219,7 @@ eval_sd, all_evals = iterate_model(model=model,
w_test=dataset_keys[dataset]['w_test'],
tau_test=None, n=n,
pred_type='predict_proba', eval_type='qini',
normalize_eval=False, notify_iter=int(n/10))
normalize_eval=False, notify_iter=n/10)

model_eval_dict[dataset].update({str(model).split('.')[-1].split(' ')[0]: {'avg_preds': avg_preds,
'all_preds': all_preds,
Expand All @@ -241,7 +242,7 @@ df_model_eval
</p>
</details>

<details><summary><strong>GRF Econometric Evaluations<strong></summary>
<details><summary><strong>GRF Econometric Evaluations (in progress)<strong></summary>
<p>

Confidence intervals are created using GRF's honesty based, Gaussian assymptotic forest summations.
Expand Down

0 comments on commit 31423e0

Please sign in to comment.