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) ```