Skip to content

Commit

Permalink
Fix order of arguments.
Browse files Browse the repository at this point in the history
  • Loading branch information
@kklein
kklein committed Oct 6, 2024
1 parent acc246c commit cbea396
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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)
```

Expand Down

0 comments on commit cbea396

Please sign in to comment.