PythonIntermediate

A tree of depth 3

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).

Train DecisionTreeClassifier(max_depth=3, random_state=0) and store the test accuracy, 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"""))