写一段使用python因果森林模型计算处理效应异质性的代码

使用python因果森林模型计算处理效应异质性的代码如下：import pandas as pd
from causalml.inference.tree import CausalForest# 读取数据
df = pd.read_csv('data.csv')# 创建因果森林模型
cf = CausalForest(n_estimators=100,
min_samples_leaf=0.05,
min_samples_split=0.05,
max_depth=10,
discrete_treatment=True,
n_jobs=-1)# 拟合模型
cf.fit(df,
treatment='Treatment',
outcome='Outcome',
controls=['Age', 'Gender'])# 计算处理效应异质性
cf.heterogeneous_effects('Treatment',
'Outcome',
controls=['Age', 'Gender'])