Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix small issues #617

Merged
merged 3 commits into from
May 23, 2022
Merged

Fix small issues #617

Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
ce805ef
Fix attribution notebook classifier usage
kbattocchi May 11, 2022
bbaec94
Limit sklearn version to avoid breaking changes
kbattocchi May 18, 2022
af7e1ee
Limit notebook refutation test to limit compute time
kbattocchi May 20, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
Fix attribution notebook classifier usage
@kbattocchi
kbattocchi committed May 19, 2022
commit ce805ef570c7e5a6fcdd708bfc128e332491ac1f
2 changes: 1 addition & 1 deletion ...os/Case Study - Multi-investment Attribution at A Software Company - EconML + DoWhy.ipynb
View file
Original file line number Diff line number Diff line change
@@ -460,7 +460,7 @@
"# initiate an EconML cate estimator\n",
"est = LinearDRLearner(\n",
" model_regression=XGBRegressor(learning_rate=0.1, max_depth=3),\n",
" model_propensity=XGBClassifier(learning_rate=0.1, max_depth=3, objective=\"multi:softmax\"),\n",
" model_propensity=XGBClassifier(learning_rate=0.1, max_depth=3, objective=\"multi:softprob\"),\n",
" random_state=1,\n",
")"
]
10 changes: 1 addition & 9 deletions ...s/CustomerScenarios/Case Study - Multi-investment Attribution at A Software Company.ipynb
View file
Original file line number Diff line number Diff line change
@@ -437,21 +437,13 @@
"# Train EconML model with generic helper models\r\n",
"model = LinearDRLearner(\r\n",
" model_regression=XGBRegressor(learning_rate=0.1, max_depth=3),\r\n",
" model_propensity=XGBClassifier(learning_rate=0.1, max_depth=3, objective=\"multi:softmax\"),\r\n",
" model_propensity=XGBClassifier(learning_rate=0.1, max_depth=3, objective=\"multi:softprob\"),\r\n",
" random_state=1,\r\n",
")\r\n",
"# Specify final stage inference type and fit model\r\n",
"model.fit(Y=Y, T=T, X=X, W=W, inference=\"statsmodels\")"
],
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"[06:14:07] WARNING: C:/Users/Administrator/workspace/xgboost-win64_release_1.3.0/src/learner.cc:1061: Starting in XGBoost 1.3.0, the default evaluation metric used with the objective 'multi:softprob' was changed from 'merror' to 'mlogloss'. Explicitly set eval_metric if you'd like to restore the old behavior.\n",
"[06:14:08] WARNING: C:/Users/Administrator/workspace/xgboost-win64_release_1.3.0/src/learner.cc:1061: Starting in XGBoost 1.3.0, the default evaluation metric used with the objective 'multi:softprob' was changed from 'merror' to 'mlogloss'. Explicitly set eval_metric if you'd like to restore the old behavior.\n"
]
},
{
"output_type": "execute_result",
"data": {