Skip to content

Commit 42b3223

Browse files
committed
Added notebook
1 parent 2208fdd commit 42b3223

File tree

1 file changed

+3
-11
lines changed

1 file changed

+3
-11
lines changed
Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
"source": [
3232
"## Set Mortality Prediction Task\n",
3333
"\n",
34-
"We use the in-hospital mortality prediction task which predicts patient mortality based on conditions and procedures."
34+
"We use the in-hospital mortality prediction task which predicts patient mortality based on diagnosis and procedure codes."
3535
]
3636
},
3737
{
@@ -117,7 +117,7 @@
117117
"\n",
118118
"trainer = Trainer(\n",
119119
" model=model,\n",
120-
" metrics=[\"roc_auc\", \"pr_auc\", \"f1\", \"accuracy\"],\n",
120+
" metrics=[\"pr_auc\", \"roc_auc\", \"f1\", \"accuracy\"],\n",
121121
")\n",
122122
"\n",
123123
"trainer.train(\n",
@@ -190,16 +190,8 @@
190190
"name": "python3"
191191
},
192192
"language_info": {
193-
"codemirror_mode": {
194-
"name": "ipython",
195-
"version": 3
196-
},
197-
"file_extension": ".py",
198-
"mimetype": "text/x-python",
199193
"name": "python",
200-
"nbconvert_exporter": "python",
201-
"pygments_lexer": "ipython3",
202-
"version": "3.10.0"
194+
"version": "3.8.0"
203195
}
204196
},
205197
"nbformat": 4,

0 commit comments

Comments
 (0)