econmlのestimatorにlgb.cvをつっこむ｜nekoumei

こんにちは。因果推論してますか？ 最近、つくりながら学ぶ! Pythonによる因果分析 を読んでてmeta-learnersいいなーって思いました。 meta-learnersは実装自体はそんなに難しくないので自力で実装してもいいんですが、個人的にはeconmlを使うのが手軽で良いです。 ※ econmlのmeta-learnersの解説、簡易な実験についてはusaitoさんの記事が分かりやすいです。 さて、そんなeconmlですが利用するestimatorはsklearnライクなapiである必要があります。 from econml.metalearners import TLearner from sklearn.ensemble import GradientBoostingRegressor est = TLearner(models=GradientBoostingRegresso