From cbea396a1ab3a91692fedc7308c663f11bf09e89 Mon Sep 17 00:00:00 2001 From: kklein Date: Sun, 6 Oct 2024 17:20:51 +0200 Subject: [PATCH] Fix order of arguments. --- 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) ```