From a4858da69a9c8b83f2c460fa0240529c75e4f6c2 Mon Sep 17 00:00:00 2001 From: Kevin Klein <7267523+kklein@users.noreply.github.com> Date: Sun, 6 Oct 2024 17:29:13 +0200 Subject: [PATCH] Fix order of arguments. (#93) --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 247df4b..a7e429e 100644 --- a/README.md +++ b/README.md @@ -34,7 +34,7 @@ rlearner = RLearner( ) features = ["age", "weight", "height"] -rlearner.fit(df[features], df["treatment"], df["outcomes"]) +rlearner.fit(df[features], df["outcomes"], df["treatment"]) cate_estimates = rlearner.predict(df[features], is_oos=False) ```