Fix order of arguments. (#93)

kklein · web-flow · commit a4858da69a9c · 2024-10-06T17:29:13.000+02:00
diff --git 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)
 ```
 

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ rlearner = RLearner(`
`34`	`34`	`)`
`35`	`35`
`36`	`36`	`features = ["age", "weight", "height"]`
`37`		`-rlearner.fit(df[features], df["treatment"], df["outcomes"])`
	`37`	`+rlearner.fit(df[features], df["outcomes"], df["treatment"])`
`38`	`38`	`cate_estimates = rlearner.predict(df[features], is_oos=False)`
`39`	`39`	```
`40`	`40`