PythonAdvanced

Balance the classes

Make 0/1 columns diabetic_flag (from diabetic), weekend_flag (from weekend_discharge) and readmit (from readmitted), 1 meaning Yes. The inputs are age, length_of_stay, medications, prior_visits, diabetic_flag and weekend_flag; the answer is readmit. Split with train_test_split (test_size=0.25, random_state=42, stratify on the answer).

Readmissions are rare, so the tree leans towards "no". Train DecisionTreeClassifier(max_depth=4, class_weight="balanced", random_state=0) and store its recall on the test part, rounded to 3 decimals, in result.

Already run for you

import io
import pandas as pd

patients = pd.read_csv(io.StringIO("""patient_id,age,length_of_stay,medications,prior_visits,diabetic,weekend_discharge,readmitted
1,42,2,9,0,No,Yes,No
2,25,10,3,0,No,No,Yes
3,39,11,1,0,No,No,No
4,66,1,12,1,No,No,No
5,40,6,6,0,Yes,No,No
6,55,7,9,0,No,Yes,No
7,68,4,8,0,Yes,No,No
8,64,1,8,0,No,No,Yes
9,28,4,3,1,No,No,No
10,33,6,3,0,No,Yes,No
11,32,7,1,0,Yes,No,No
12,68,14,7,0,Yes,No,No
13,64,5,11,0,No,No,No
14,48,1,6,3,No,No,No
15,60,5,10,0,No,No,Yes
16,71,5,3,0,Yes,No,No
17,71,2,10,2,No,Yes,No
18,58,3,4,0,No,Yes,No
19,67,3,6,0,Yes,Yes,No
20,39,7,2,0,Yes,No,No
21,51,1,7,1,Yes,Yes,No
22,45,5,9,0,Yes,No,No
23,69,6,6,1,No,No,No
24,50,7,1,0,Yes,Yes,No
25,55,1,8,1,No,No,No
26,83,3,5,0,No,No,No
27,75,1,10,0,Yes,No,No
28,54,1,3,0,No,No,No
29,41,3,5,1,Yes,No,No
30,72,9,6,0,Yes,Yes,No
31,74,2,10,0,No,Yes,No
32,71,4,4,0,No,No,No
33,78,1,9,0,Yes,No,Yes
34,67,9,1,0,No,No,No
35,64,5,7,0,No,No,No
36,56,2,9,0,No,No,No
37,28,12,4,0,Yes,Yes,No
38,57,5,7,2,No,Yes,Yes
39,81,8,10,0,Yes,No,No
40,64,8,9,0,No,No,No
41,29,4,3,8,Yes,No,No
42,48,5,6,0,Yes,Yes,No
43,63,5,5,0,Yes,Yes,No
44,41,3,8,0,No,No,No
45,72,1,3,1,No,Yes,No
46,63,4,5,1,No,No,No
47,50,5,9,4,No,No,Yes
48,31,11,4,0,No,Yes,No
49,62,4,5,0,No,No,No
50,53,1,7,1,Yes,No,No
51,59,1,2,1,Yes,No,No
52,78,7,6,0,No,No,Yes
53,45,4,6,0,Yes,Yes,No
54,43,1,8,5,No,No,No
55,80,2,9,3,No,No,No
56,64,3,8,2,Yes,Yes,No
57,70,4,6,0,No,No,No
58,78,3,7,0,Yes,No,No
59,35,9,5,0,No,No,No
60,42,3,8,0,No,No,No
61,40,1,3,0,No,No,No
62,69,6,6,0,No,No,No
63,38,3,9,0,Yes,No,No
64,80,7,11,0,Yes,No,No
65,48,4,10,0,Yes,No,No
66,30,6,2,1,Yes,Yes,No
67,41,1,6,0,Yes,No,No
68,32,1,5,0,No,No,Yes
69,47,1,9,0,No,Yes,Yes
70,71,2,8,0,Yes,Yes,No
71,40,18,7,0,No,No,No
72,69,2,10,4,Yes,No,No
73,55,3,8,0,Yes,No,No
74,38,2,5,0,No,No,No
75,52,3,9,0,No,No,No
76,36,19,2,0,Yes,No,No
77,50,2,9,0,No,Yes,No
78,32,5,4,0,No,Yes,No
79,33,4,4,3,No,Yes,No
80,32,20,6,0,No,No,No
81,47,6,8,1,Yes,No,No
82,45,1,2,0,No,No,No
83,55,2,7,0,No,Yes,No
84,40,9,6,3,No,No,No
85,39,1,4,5,No,No,Yes
86,28,2,4,2,No,No,No
87,83,2,7,0,No,No,No
88,47,2,7,3,Yes,No,Yes
89,57,2,6,0,No,No,No
90,70,11,17,0,No,No,No
91,42,2,8,0,Yes,No,No
92,47,6,7,0,Yes,No,No
93,73,2,11,0,No,No,No
94,40,5,9,0,Yes,Yes,No
95,64,7,8,0,No,No,No
96,37,1,2,0,No,Yes,No
97,26,4,7,0,No,No,No
98,49,6,1,0,No,No,No
99,76,1,10,6,Yes,No,No
100,46,3,13,0,Yes,Yes,No
101,66,5,12,1,Yes,No,No
102,76,3,8,0,Yes,No,Yes
103,59,1,6,0,No,No,No
104,50,6,5,0,No,No,No
105,61,7,9,0,No,No,No
106,83,1,13,4,No,Yes,Yes
107,39,10,5,0,Yes,No,No
108,33,1,6,1,No,No,No
109,53,3,6,0,No,No,No
110,52,3,4,0,No,No,No
111,31,1,5,2,No,No,Yes
112,42,1,3,0,Yes,No,No
113,39,1,12,0,Yes,Yes,No
114,84,7,8,0,Yes,Yes,No
115,83,4,14,0,Yes,No,Yes
116,61,1,7,0,No,No,No
117,81,4,5,0,Yes,No,No
118,55,3,1,0,Yes,No,No
119,57,2,7,1,Yes,Yes,Yes
120,68,2,11,1,Yes,No,No
121,50,13,5,0,No,No,No
122,31,2,4,0,No,No,No
123,39,2,5,2,No,No,No
124,71,4,9,2,No,No,No
125,78,1,2,0,No,No,No
126,41,1,6,0,No,Yes,Yes
127,29,3,1,2,Yes,No,No
128,71,1,10,4,Yes,No,Yes
129,81,1,8,0,Yes,Yes,No
130,62,9,13,2,Yes,No,Yes
131,57,3,9,0,No,No,No
132,44,6,6,0,No,No,No
133,33,6,11,2,No,No,No
134,63,4,4,0,No,Yes,No
135,66,2,6,0,No,Yes,No
136,44,1,4,2,Yes,No,No
137,77,10,6,0,Yes,Yes,No
138,83,1,10,1,No,No,No
139,42,12,9,0,No,No,No
140,29,5,3,3,No,No,Yes
141,50,5,2,0,No,No,No
142,40,7,1,2,No,No,No
143,51,5,6,0,No,No,No
144,32,1,1,0,No,No,No
145,82,5,5,3,No,Yes,No
146,49,10,9,0,No,No,No
147,65,12,9,1,Yes,Yes,No
148,81,6,6,0,Yes,Yes,No
149,79,5,9,0,No,Yes,No
150,35,3,11,0,No,No,No
151,60,1,4,0,No,Yes,Yes
152,84,4,8,1,Yes,No,No
153,66,2,5,0,Yes,No,No
154,69,2,4,1,No,No,No
155,49,3,11,6,No,No,Yes
156,48,2,5,0,Yes,No,No
157,44,17,3,3,No,No,No
158,65,3,7,0,No,No,No
159,36,1,2,0,No,Yes,Yes
160,78,5,7,8,No,No,No
161,60,2,9,0,No,No,No
162,41,1,5,0,Yes,Yes,No
163,44,2,2,0,Yes,No,No
164,45,1,1,7,No,Yes,Yes
165,34,15,2,0,No,No,No
166,79,3,4,0,No,No,No
167,41,4,12,0,No,No,No
168,74,1,11,0,No,No,No
169,81,3,13,0,No,No,Yes
170,37,2,2,0,No,No,No
171,45,2,6,0,No,No,No
172,33,1,6,4,Yes,Yes,Yes
173,39,3,4,2,No,No,No
174,38,13,5,0,No,Yes,No
175,70,5,6,0,No,No,No
176,77,8,8,0,Yes,No,No
177,57,6,5,0,No,Yes,No
178,38,5,5,0,No,Yes,Yes
179,55,1,10,0,No,No,No
180,68,2,5,1,Yes,No,No
181,59,8,5,7,No,No,Yes
182,63,6,8,0,No,No,No
183,37,14,6,0,No,No,No
184,40,9,3,0,No,No,No
185,79,1,7,1,No,No,No
186,73,4,8,0,Yes,No,No
187,34,2,7,0,No,No,No
188,45,13,6,0,No,No,No
189,54,1,10,2,No,No,No
190,48,3,14,0,No,No,Yes
191,49,5,10,0,No,No,No
192,34,7,9,1,Yes,Yes,No
193,50,8,6,0,No,No,No
194,49,1,11,0,No,Yes,No
195,28,5,1,0,Yes,No,No
196,48,5,8,0,No,No,No
197,26,2,5,0,No,No,No
198,60,4,8,0,Yes,No,No
199,33,3,5,0,Yes,No,No
200,43,6,8,0,Yes,No,Yes
201,78,10,14,0,No,No,No
202,61,5,11,0,No,No,No
203,35,4,1,0,No,Yes,No
204,63,10,7,0,No,No,No
205,55,7,5,0,No,No,No
206,77,3,8,0,Yes,Yes,No
207,83,1,8,0,No,No,No
208,81,6,8,0,Yes,No,No
209,53,7,11,0,Yes,No,No
210,60,9,7,5,No,No,No
211,65,7,11,0,No,No,Yes
212,83,6,9,0,No,No,No
213,58,1,6,0,No,No,No
214,48,1,7,0,No,No,No
215,83,3,8,1,Yes,No,No
216,41,7,3,0,Yes,No,No
217,27,5,10,4,No,No,No
218,67,2,5,2,No,No,No
219,37,2,5,3,No,No,No
220,34,14,6,1,Yes,No,Yes
221,32,6,4,0,No,No,No
222,53,1,9,0,No,No,No
223,37,8,7,0,No,No,No
224,74,4,7,0,No,No,No
225,30,17,5,4,No,Yes,Yes
226,33,3,5,3,Yes,No,Yes
227,39,6,3,0,Yes,No,No
228,78,6,5,2,Yes,No,No
229,32,3,6,0,Yes,No,No
230,73,2,9,1,No,No,No
231,68,1,6,1,No,No,No
232,73,2,9,0,Yes,No,Yes
233,78,9,5,0,Yes,No,Yes
234,66,3,6,3,Yes,No,No
235,25,3,2,0,No,Yes,No
236,45,3,13,0,Yes,No,Yes
237,37,11,2,0,Yes,No,No
238,46,2,5,0,No,Yes,No
239,45,3,2,0,Yes,No,No
240,76,1,11,0,No,No,No
241,31,2,6,0,No,No,No
242,68,9,8,0,No,No,No
243,29,2,8,0,No,No,No
244,48,3,2,5,Yes,No,No
245,35,5,5,1,No,Yes,No
246,75,4,1,5,Yes,No,No
247,56,3,3,2,No,No,No
248,49,16,5,0,No,No,No
249,69,1,5,0,Yes,No,No
250,77,4,7,0,No,Yes,No
251,69,2,1,0,No,No,No
252,25,9,1,0,No,No,No
253,62,4,11,0,No,No,No
254,61,6,5,0,Yes,No,No
255,75,5,10,0,No,Yes,No
256,29,6,8,0,No,No,Yes
257,73,3,10,4,Yes,No,Yes
258,74,4,12,1,No,No,No
259,62,17,4,0,No,No,No
260,59,1,14,0,No,Yes,No"""))

Store your final answer in a variable named result.

Hints(0/2)

Each hint gives a little more help. Try on your own first.

Solution

Unlocks after you solve it or after 3 submissions

Notebook · scikit-learnRun and Submit check all cells from the top

Shift + Enter30 seconds limit per cell

[ ]

Run your code to see the output here.

Already run for you

import io import pandas as pd patients = pd.read_csv(io.StringIO("""patient_id,age,length_of_stay,medications,prior_visits,diabetic,weekend_discharge,readmitted 1,42,2,9,0,No,Yes,No 2,25,10,3,0,No,No,Yes 3,39,11,1,0,No,No,No 4,66,1,12,1,No,No,No 5,40,6,6,0,Yes,No,No 6,55,7,9,0,No,Yes,No 7,68,4,8,0,Yes,No,No 8,64,1,8,0,No,No,Yes 9,28,4,3,1,No,No,No 10,33,6,3,0,No,Yes,No 11,32,7,1,0,Yes,No,No 12,68,14,7,0,Yes,No,No 13,64,5,11,0,No,No,No 14,48,1,6,3,No,No,No 15,60,5,10,0,No,No,Yes 16,71,5,3,0,Yes,No,No 17,71,2,10,2,No,Yes,No 18,58,3,4,0,No,Yes,No 19,67,3,6,0,Yes,Yes,No 20,39,7,2,0,Yes,No,No 21,51,1,7,1,Yes,Yes,No 22,45,5,9,0,Yes,No,No 23,69,6,6,1,No,No,No 24,50,7,1,0,Yes,Yes,No 25,55,1,8,1,No,No,No 26,83,3,5,0,No,No,No 27,75,1,10,0,Yes,No,No 28,54,1,3,0,No,No,No 29,41,3,5,1,Yes,No,No 30,72,9,6,0,Yes,Yes,No 31,74,2,10,0,No,Yes,No 32,71,4,4,0,No,No,No 33,78,1,9,0,Yes,No,Yes 34,67,9,1,0,No,No,No 35,64,5,7,0,No,No,No 36,56,2,9,0,No,No,No 37,28,12,4,0,Yes,Yes,No 38,57,5,7,2,No,Yes,Yes 39,81,8,10,0,Yes,No,No 40,64,8,9,0,No,No,No 41,29,4,3,8,Yes,No,No 42,48,5,6,0,Yes,Yes,No 43,63,5,5,0,Yes,Yes,No 44,41,3,8,0,No,No,No 45,72,1,3,1,No,Yes,No 46,63,4,5,1,No,No,No 47,50,5,9,4,No,No,Yes 48,31,11,4,0,No,Yes,No 49,62,4,5,0,No,No,No 50,53,1,7,1,Yes,No,No 51,59,1,2,1,Yes,No,No 52,78,7,6,0,No,No,Yes 53,45,4,6,0,Yes,Yes,No 54,43,1,8,5,No,No,No 55,80,2,9,3,No,No,No 56,64,3,8,2,Yes,Yes,No 57,70,4,6,0,No,No,No 58,78,3,7,0,Yes,No,No 59,35,9,5,0,No,No,No 60,42,3,8,0,No,No,No 61,40,1,3,0,No,No,No 62,69,6,6,0,No,No,No 63,38,3,9,0,Yes,No,No 64,80,7,11,0,Yes,No,No 65,48,4,10,0,Yes,No,No 66,30,6,2,1,Yes,Yes,No 67,41,1,6,0,Yes,No,No 68,32,1,5,0,No,No,Yes 69,47,1,9,0,No,Yes,Yes 70,71,2,8,0,Yes,Yes,No 71,40,18,7,0,No,No,No 72,69,2,10,4,Yes,No,No 73,55,3,8,0,Yes,No,No 74,38,2,5,0,No,No,No 75,52,3,9,0,No,No,No 76,36,19,2,0,Yes,No,No 77,50,2,9,0,No,Yes,No 78,32,5,4,0,No,Yes,No 79,33,4,4,3,No,Yes,No 80,32,20,6,0,No,No,No 81,47,6,8,1,Yes,No,No 82,45,1,2,0,No,No,No 83,55,2,7,0,No,Yes,No 84,40,9,6,3,No,No,No 85,39,1,4,5,No,No,Yes 86,28,2,4,2,No,No,No 87,83,2,7,0,No,No,No 88,47,2,7,3,Yes,No,Yes 89,57,2,6,0,No,No,No 90,70,11,17,0,No,No,No 91,42,2,8,0,Yes,No,No 92,47,6,7,0,Yes,No,No 93,73,2,11,0,No,No,No 94,40,5,9,0,Yes,Yes,No 95,64,7,8,0,No,No,No 96,37,1,2,0,No,Yes,No 97,26,4,7,0,No,No,No 98,49,6,1,0,No,No,No 99,76,1,10,6,Yes,No,No 100,46,3,13,0,Yes,Yes,No 101,66,5,12,1,Yes,No,No 102,76,3,8,0,Yes,No,Yes 103,59,1,6,0,No,No,No 104,50,6,5,0,No,No,No 105,61,7,9,0,No,No,No 106,83,1,13,4,No,Yes,Yes 107,39,10,5,0,Yes,No,No 108,33,1,6,1,No,No,No 109,53,3,6,0,No,No,No 110,52,3,4,0,No,No,No 111,31,1,5,2,No,No,Yes 112,42,1,3,0,Yes,No,No 113,39,1,12,0,Yes,Yes,No 114,84,7,8,0,Yes,Yes,No 115,83,4,14,0,Yes,No,Yes 116,61,1,7,0,No,No,No 117,81,4,5,0,Yes,No,No 118,55,3,1,0,Yes,No,No 119,57,2,7,1,Yes,Yes,Yes 120,68,2,11,1,Yes,No,No 121,50,13,5,0,No,No,No 122,31,2,4,0,No,No,No 123,39,2,5,2,No,No,No 124,71,4,9,2,No,No,No 125,78,1,2,0,No,No,No 126,41,1,6,0,No,Yes,Yes 127,29,3,1,2,Yes,No,No 128,71,1,10,4,Yes,No,Yes 129,81,1,8,0,Yes,Yes,No 130,62,9,13,2,Yes,No,Yes 131,57,3,9,0,No,No,No 132,44,6,6,0,No,No,No 133,33,6,11,2,No,No,No 134,63,4,4,0,No,Yes,No 135,66,2,6,0,No,Yes,No 136,44,1,4,2,Yes,No,No 137,77,10,6,0,Yes,Yes,No 138,83,1,10,1,No,No,No 139,42,12,9,0,No,No,No 140,29,5,3,3,No,No,Yes 141,50,5,2,0,No,No,No 142,40,7,1,2,No,No,No 143,51,5,6,0,No,No,No 144,32,1,1,0,No,No,No 145,82,5,5,3,No,Yes,No 146,49,10,9,0,No,No,No 147,65,12,9,1,Yes,Yes,No 148,81,6,6,0,Yes,Yes,No 149,79,5,9,0,No,Yes,No 150,35,3,11,0,No,No,No 151,60,1,4,0,No,Yes,Yes 152,84,4,8,1,Yes,No,No 153,66,2,5,0,Yes,No,No 154,69,2,4,1,No,No,No 155,49,3,11,6,No,No,Yes 156,48,2,5,0,Yes,No,No 157,44,17,3,3,No,No,No 158,65,3,7,0,No,No,No 159,36,1,2,0,No,Yes,Yes 160,78,5,7,8,No,No,No 161,60,2,9,0,No,No,No 162,41,1,5,0,Yes,Yes,No 163,44,2,2,0,Yes,No,No 164,45,1,1,7,No,Yes,Yes 165,34,15,2,0,No,No,No 166,79,3,4,0,No,No,No 167,41,4,12,0,No,No,No 168,74,1,11,0,No,No,No 169,81,3,13,0,No,No,Yes 170,37,2,2,0,No,No,No 171,45,2,6,0,No,No,No 172,33,1,6,4,Yes,Yes,Yes 173,39,3,4,2,No,No,No 174,38,13,5,0,No,Yes,No 175,70,5,6,0,No,No,No 176,77,8,8,0,Yes,No,No 177,57,6,5,0,No,Yes,No 178,38,5,5,0,No,Yes,Yes 179,55,1,10,0,No,No,No 180,68,2,5,1,Yes,No,No 181,59,8,5,7,No,No,Yes 182,63,6,8,0,No,No,No 183,37,14,6,0,No,No,No 184,40,9,3,0,No,No,No 185,79,1,7,1,No,No,No 186,73,4,8,0,Yes,No,No 187,34,2,7,0,No,No,No 188,45,13,6,0,No,No,No 189,54,1,10,2,No,No,No 190,48,3,14,0,No,No,Yes 191,49,5,10,0,No,No,No 192,34,7,9,1,Yes,Yes,No 193,50,8,6,0,No,No,No 194,49,1,11,0,No,Yes,No 195,28,5,1,0,Yes,No,No 196,48,5,8,0,No,No,No 197,26,2,5,0,No,No,No 198,60,4,8,0,Yes,No,No 199,33,3,5,0,Yes,No,No 200,43,6,8,0,Yes,No,Yes 201,78,10,14,0,No,No,No 202,61,5,11,0,No,No,No 203,35,4,1,0,No,Yes,No 204,63,10,7,0,No,No,No 205,55,7,5,0,No,No,No 206,77,3,8,0,Yes,Yes,No 207,83,1,8,0,No,No,No 208,81,6,8,0,Yes,No,No 209,53,7,11,0,Yes,No,No 210,60,9,7,5,No,No,No 211,65,7,11,0,No,No,Yes 212,83,6,9,0,No,No,No 213,58,1,6,0,No,No,No 214,48,1,7,0,No,No,No 215,83,3,8,1,Yes,No,No 216,41,7,3,0,Yes,No,No 217,27,5,10,4,No,No,No 218,67,2,5,2,No,No,No 219,37,2,5,3,No,No,No 220,34,14,6,1,Yes,No,Yes 221,32,6,4,0,No,No,No 222,53,1,9,0,No,No,No 223,37,8,7,0,No,No,No 224,74,4,7,0,No,No,No 225,30,17,5,4,No,Yes,Yes 226,33,3,5,3,Yes,No,Yes 227,39,6,3,0,Yes,No,No 228,78,6,5,2,Yes,No,No 229,32,3,6,0,Yes,No,No 230,73,2,9,1,No,No,No 231,68,1,6,1,No,No,No 232,73,2,9,0,Yes,No,Yes 233,78,9,5,0,Yes,No,Yes 234,66,3,6,3,Yes,No,No 235,25,3,2,0,No,Yes,No 236,45,3,13,0,Yes,No,Yes 237,37,11,2,0,Yes,No,No 238,46,2,5,0,No,Yes,No 239,45,3,2,0,Yes,No,No 240,76,1,11,0,No,No,No 241,31,2,6,0,No,No,No 242,68,9,8,0,No,No,No 243,29,2,8,0,No,No,No 244,48,3,2,5,Yes,No,No 245,35,5,5,1,No,Yes,No 246,75,4,1,5,Yes,No,No 247,56,3,3,2,No,No,No 248,49,16,5,0,No,No,No 249,69,1,5,0,Yes,No,No 250,77,4,7,0,No,Yes,No 251,69,2,1,0,No,No,No 252,25,9,1,0,No,No,No 253,62,4,11,0,No,No,No 254,61,6,5,0,Yes,No,No 255,75,5,10,0,No,Yes,No 256,29,6,8,0,No,No,Yes 257,73,3,10,4,Yes,No,Yes 258,74,4,12,1,No,No,No 259,62,17,4,0,No,No,No 260,59,1,14,0,No,Yes,No"""))