Hugging Face's logo

StudentDepressionPrediction
/
Student_Depression_Prediction

Model card Files
xet
 Community
Boorla commited on
Commit
0ff646c
·
verified ·
1 Parent(s): ac29cfb

Upload 3 files

Browse files
Files changed (3) hide show
  1. Student_Depresion_Prediction.ipynb +2173 -0
  2. app.py +68 -0
  3. student_depression_dataset.csv +0 -0
Student_Depresion_Prediction.ipynb ADDED
@@ -0,0 +1,2173 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cells": [
3
+ {
4
+ "cell_type": "code",
5
+ "execution_count": 93,
6
+ "id": "8a95934e-ff3f-41b1-be3a-c63b4f625c9e",
7
+ "metadata": {},
8
+ "outputs": [],
9
+ "source": [
10
+ "import pandas as pd"
11
+ ]
12
+ },
13
+ {
14
+ "cell_type": "code",
15
+ "execution_count": 95,
16
+ "id": "9cf17eea-09e2-4e51-9c59-bd13453b81cd",
17
+ "metadata": {},
18
+ "outputs": [
19
+ {
20
+ "data": {
21
+ "text/html": [
22
+ "<div>\n",
23
+ "<style scoped>\n",
24
+ " .dataframe tbody tr th:only-of-type {\n",
25
+ " vertical-align: middle;\n",
26
+ " }\n",
27
+ "\n",
28
+ " .dataframe tbody tr th {\n",
29
+ " vertical-align: top;\n",
30
+ " }\n",
31
+ "\n",
32
+ " .dataframe thead th {\n",
33
+ " text-align: right;\n",
34
+ " }\n",
35
+ "</style>\n",
36
+ "<table border=\"1\" class=\"dataframe\">\n",
37
+ " <thead>\n",
38
+ " <tr style=\"text-align: right;\">\n",
39
+ " <th></th>\n",
40
+ " <th>id</th>\n",
41
+ " <th>Gender</th>\n",
42
+ " <th>Age</th>\n",
43
+ " <th>City</th>\n",
44
+ " <th>Profession</th>\n",
45
+ " <th>Academic Pressure</th>\n",
46
+ " <th>Work Pressure</th>\n",
47
+ " <th>CGPA</th>\n",
48
+ " <th>Study Satisfaction</th>\n",
49
+ " <th>Job Satisfaction</th>\n",
50
+ " <th>Sleep Duration</th>\n",
51
+ " <th>Dietary Habits</th>\n",
52
+ " <th>Degree</th>\n",
53
+ " <th>Have you ever had suicidal thoughts ?</th>\n",
54
+ " <th>Work/Study Hours</th>\n",
55
+ " <th>Financial Stress</th>\n",
56
+ " <th>Family History of Mental Illness</th>\n",
57
+ " <th>Depression</th>\n",
58
+ " </tr>\n",
59
+ " </thead>\n",
60
+ " <tbody>\n",
61
+ " <tr>\n",
62
+ " <th>0</th>\n",
63
+ " <td>2</td>\n",
64
+ " <td>Male</td>\n",
65
+ " <td>33.0</td>\n",
66
+ " <td>Visakhapatnam</td>\n",
67
+ " <td>Student</td>\n",
68
+ " <td>5.0</td>\n",
69
+ " <td>0.0</td>\n",
70
+ " <td>8.97</td>\n",
71
+ " <td>2.0</td>\n",
72
+ " <td>0.0</td>\n",
73
+ " <td>'5-6 hours'</td>\n",
74
+ " <td>Healthy</td>\n",
75
+ " <td>B.Pharm</td>\n",
76
+ " <td>Yes</td>\n",
77
+ " <td>3.0</td>\n",
78
+ " <td>1.0</td>\n",
79
+ " <td>No</td>\n",
80
+ " <td>1</td>\n",
81
+ " </tr>\n",
82
+ " <tr>\n",
83
+ " <th>1</th>\n",
84
+ " <td>8</td>\n",
85
+ " <td>Female</td>\n",
86
+ " <td>24.0</td>\n",
87
+ " <td>Bangalore</td>\n",
88
+ " <td>Student</td>\n",
89
+ " <td>2.0</td>\n",
90
+ " <td>0.0</td>\n",
91
+ " <td>5.90</td>\n",
92
+ " <td>5.0</td>\n",
93
+ " <td>0.0</td>\n",
94
+ " <td>'5-6 hours'</td>\n",
95
+ " <td>Moderate</td>\n",
96
+ " <td>BSc</td>\n",
97
+ " <td>No</td>\n",
98
+ " <td>3.0</td>\n",
99
+ " <td>2.0</td>\n",
100
+ " <td>Yes</td>\n",
101
+ " <td>0</td>\n",
102
+ " </tr>\n",
103
+ " <tr>\n",
104
+ " <th>2</th>\n",
105
+ " <td>26</td>\n",
106
+ " <td>Male</td>\n",
107
+ " <td>31.0</td>\n",
108
+ " <td>Srinagar</td>\n",
109
+ " <td>Student</td>\n",
110
+ " <td>3.0</td>\n",
111
+ " <td>0.0</td>\n",
112
+ " <td>7.03</td>\n",
113
+ " <td>5.0</td>\n",
114
+ " <td>0.0</td>\n",
115
+ " <td>'Less than 5 hours'</td>\n",
116
+ " <td>Healthy</td>\n",
117
+ " <td>BA</td>\n",
118
+ " <td>No</td>\n",
119
+ " <td>9.0</td>\n",
120
+ " <td>1.0</td>\n",
121
+ " <td>Yes</td>\n",
122
+ " <td>0</td>\n",
123
+ " </tr>\n",
124
+ " <tr>\n",
125
+ " <th>3</th>\n",
126
+ " <td>30</td>\n",
127
+ " <td>Female</td>\n",
128
+ " <td>28.0</td>\n",
129
+ " <td>Varanasi</td>\n",
130
+ " <td>Student</td>\n",
131
+ " <td>3.0</td>\n",
132
+ " <td>0.0</td>\n",
133
+ " <td>5.59</td>\n",
134
+ " <td>2.0</td>\n",
135
+ " <td>0.0</td>\n",
136
+ " <td>'7-8 hours'</td>\n",
137
+ " <td>Moderate</td>\n",
138
+ " <td>BCA</td>\n",
139
+ " <td>Yes</td>\n",
140
+ " <td>4.0</td>\n",
141
+ " <td>5.0</td>\n",
142
+ " <td>Yes</td>\n",
143
+ " <td>1</td>\n",
144
+ " </tr>\n",
145
+ " <tr>\n",
146
+ " <th>4</th>\n",
147
+ " <td>32</td>\n",
148
+ " <td>Female</td>\n",
149
+ " <td>25.0</td>\n",
150
+ " <td>Jaipur</td>\n",
151
+ " <td>Student</td>\n",
152
+ " <td>4.0</td>\n",
153
+ " <td>0.0</td>\n",
154
+ " <td>8.13</td>\n",
155
+ " <td>3.0</td>\n",
156
+ " <td>0.0</td>\n",
157
+ " <td>'5-6 hours'</td>\n",
158
+ " <td>Moderate</td>\n",
159
+ " <td>M.Tech</td>\n",
160
+ " <td>Yes</td>\n",
161
+ " <td>1.0</td>\n",
162
+ " <td>1.0</td>\n",
163
+ " <td>No</td>\n",
164
+ " <td>0</td>\n",
165
+ " </tr>\n",
166
+ " <tr>\n",
167
+ " <th>...</th>\n",
168
+ " <td>...</td>\n",
169
+ " <td>...</td>\n",
170
+ " <td>...</td>\n",
171
+ " <td>...</td>\n",
172
+ " <td>...</td>\n",
173
+ " <td>...</td>\n",
174
+ " <td>...</td>\n",
175
+ " <td>...</td>\n",
176
+ " <td>...</td>\n",
177
+ " <td>...</td>\n",
178
+ " <td>...</td>\n",
179
+ " <td>...</td>\n",
180
+ " <td>...</td>\n",
181
+ " <td>...</td>\n",
182
+ " <td>...</td>\n",
183
+ " <td>...</td>\n",
184
+ " <td>...</td>\n",
185
+ " <td>...</td>\n",
186
+ " </tr>\n",
187
+ " <tr>\n",
188
+ " <th>27896</th>\n",
189
+ " <td>140685</td>\n",
190
+ " <td>Female</td>\n",
191
+ " <td>27.0</td>\n",
192
+ " <td>Surat</td>\n",
193
+ " <td>Student</td>\n",
194
+ " <td>5.0</td>\n",
195
+ " <td>0.0</td>\n",
196
+ " <td>5.75</td>\n",
197
+ " <td>5.0</td>\n",
198
+ " <td>0.0</td>\n",
199
+ " <td>'5-6 hours'</td>\n",
200
+ " <td>Unhealthy</td>\n",
201
+ " <td>'Class 12'</td>\n",
202
+ " <td>Yes</td>\n",
203
+ " <td>7.0</td>\n",
204
+ " <td>1.0</td>\n",
205
+ " <td>Yes</td>\n",
206
+ " <td>0</td>\n",
207
+ " </tr>\n",
208
+ " <tr>\n",
209
+ " <th>27897</th>\n",
210
+ " <td>140686</td>\n",
211
+ " <td>Male</td>\n",
212
+ " <td>27.0</td>\n",
213
+ " <td>Ludhiana</td>\n",
214
+ " <td>Student</td>\n",
215
+ " <td>2.0</td>\n",
216
+ " <td>0.0</td>\n",
217
+ " <td>9.40</td>\n",
218
+ " <td>3.0</td>\n",
219
+ " <td>0.0</td>\n",
220
+ " <td>'Less than 5 hours'</td>\n",
221
+ " <td>Healthy</td>\n",
222
+ " <td>MSc</td>\n",
223
+ " <td>No</td>\n",
224
+ " <td>0.0</td>\n",
225
+ " <td>3.0</td>\n",
226
+ " <td>Yes</td>\n",
227
+ " <td>0</td>\n",
228
+ " </tr>\n",
229
+ " <tr>\n",
230
+ " <th>27898</th>\n",
231
+ " <td>140689</td>\n",
232
+ " <td>Male</td>\n",
233
+ " <td>31.0</td>\n",
234
+ " <td>Faridabad</td>\n",
235
+ " <td>Student</td>\n",
236
+ " <td>3.0</td>\n",
237
+ " <td>0.0</td>\n",
238
+ " <td>6.61</td>\n",
239
+ " <td>4.0</td>\n",
240
+ " <td>0.0</td>\n",
241
+ " <td>'5-6 hours'</td>\n",
242
+ " <td>Unhealthy</td>\n",
243
+ " <td>MD</td>\n",
244
+ " <td>No</td>\n",
245
+ " <td>12.0</td>\n",
246
+ " <td>2.0</td>\n",
247
+ " <td>No</td>\n",
248
+ " <td>0</td>\n",
249
+ " </tr>\n",
250
+ " <tr>\n",
251
+ " <th>27899</th>\n",
252
+ " <td>140690</td>\n",
253
+ " <td>Female</td>\n",
254
+ " <td>18.0</td>\n",
255
+ " <td>Ludhiana</td>\n",
256
+ " <td>Student</td>\n",
257
+ " <td>5.0</td>\n",
258
+ " <td>0.0</td>\n",
259
+ " <td>6.88</td>\n",
260
+ " <td>2.0</td>\n",
261
+ " <td>0.0</td>\n",
262
+ " <td>'Less than 5 hours'</td>\n",
263
+ " <td>Healthy</td>\n",
264
+ " <td>'Class 12'</td>\n",
265
+ " <td>Yes</td>\n",
266
+ " <td>10.0</td>\n",
267
+ " <td>5.0</td>\n",
268
+ " <td>No</td>\n",
269
+ " <td>1</td>\n",
270
+ " </tr>\n",
271
+ " <tr>\n",
272
+ " <th>27900</th>\n",
273
+ " <td>140699</td>\n",
274
+ " <td>Male</td>\n",
275
+ " <td>27.0</td>\n",
276
+ " <td>Patna</td>\n",
277
+ " <td>Student</td>\n",
278
+ " <td>4.0</td>\n",
279
+ " <td>0.0</td>\n",
280
+ " <td>9.24</td>\n",
281
+ " <td>1.0</td>\n",
282
+ " <td>0.0</td>\n",
283
+ " <td>'Less than 5 hours'</td>\n",
284
+ " <td>Healthy</td>\n",
285
+ " <td>BCA</td>\n",
286
+ " <td>Yes</td>\n",
287
+ " <td>2.0</td>\n",
288
+ " <td>3.0</td>\n",
289
+ " <td>Yes</td>\n",
290
+ " <td>1</td>\n",
291
+ " </tr>\n",
292
+ " </tbody>\n",
293
+ "</table>\n",
294
+ "<p>27901 rows × 18 columns</p>\n",
295
+ "</div>"
296
+ ],
297
+ "text/plain": [
298
+ " id Gender Age City Profession Academic Pressure \\\n",
299
+ "0 2 Male 33.0 Visakhapatnam Student 5.0 \n",
300
+ "1 8 Female 24.0 Bangalore Student 2.0 \n",
301
+ "2 26 Male 31.0 Srinagar Student 3.0 \n",
302
+ "3 30 Female 28.0 Varanasi Student 3.0 \n",
303
+ "4 32 Female 25.0 Jaipur Student 4.0 \n",
304
+ "... ... ... ... ... ... ... \n",
305
+ "27896 140685 Female 27.0 Surat Student 5.0 \n",
306
+ "27897 140686 Male 27.0 Ludhiana Student 2.0 \n",
307
+ "27898 140689 Male 31.0 Faridabad Student 3.0 \n",
308
+ "27899 140690 Female 18.0 Ludhiana Student 5.0 \n",
309
+ "27900 140699 Male 27.0 Patna Student 4.0 \n",
310
+ "\n",
311
+ " Work Pressure CGPA Study Satisfaction Job Satisfaction \\\n",
312
+ "0 0.0 8.97 2.0 0.0 \n",
313
+ "1 0.0 5.90 5.0 0.0 \n",
314
+ "2 0.0 7.03 5.0 0.0 \n",
315
+ "3 0.0 5.59 2.0 0.0 \n",
316
+ "4 0.0 8.13 3.0 0.0 \n",
317
+ "... ... ... ... ... \n",
318
+ "27896 0.0 5.75 5.0 0.0 \n",
319
+ "27897 0.0 9.40 3.0 0.0 \n",
320
+ "27898 0.0 6.61 4.0 0.0 \n",
321
+ "27899 0.0 6.88 2.0 0.0 \n",
322
+ "27900 0.0 9.24 1.0 0.0 \n",
323
+ "\n",
324
+ " Sleep Duration Dietary Habits Degree \\\n",
325
+ "0 '5-6 hours' Healthy B.Pharm \n",
326
+ "1 '5-6 hours' Moderate BSc \n",
327
+ "2 'Less than 5 hours' Healthy BA \n",
328
+ "3 '7-8 hours' Moderate BCA \n",
329
+ "4 '5-6 hours' Moderate M.Tech \n",
330
+ "... ... ... ... \n",
331
+ "27896 '5-6 hours' Unhealthy 'Class 12' \n",
332
+ "27897 'Less than 5 hours' Healthy MSc \n",
333
+ "27898 '5-6 hours' Unhealthy MD \n",
334
+ "27899 'Less than 5 hours' Healthy 'Class 12' \n",
335
+ "27900 'Less than 5 hours' Healthy BCA \n",
336
+ "\n",
337
+ " Have you ever had suicidal thoughts ? Work/Study Hours \\\n",
338
+ "0 Yes 3.0 \n",
339
+ "1 No 3.0 \n",
340
+ "2 No 9.0 \n",
341
+ "3 Yes 4.0 \n",
342
+ "4 Yes 1.0 \n",
343
+ "... ... ... \n",
344
+ "27896 Yes 7.0 \n",
345
+ "27897 No 0.0 \n",
346
+ "27898 No 12.0 \n",
347
+ "27899 Yes 10.0 \n",
348
+ "27900 Yes 2.0 \n",
349
+ "\n",
350
+ " Financial Stress Family History of Mental Illness Depression \n",
351
+ "0 1.0 No 1 \n",
352
+ "1 2.0 Yes 0 \n",
353
+ "2 1.0 Yes 0 \n",
354
+ "3 5.0 Yes 1 \n",
355
+ "4 1.0 No 0 \n",
356
+ "... ... ... ... \n",
357
+ "27896 1.0 Yes 0 \n",
358
+ "27897 3.0 Yes 0 \n",
359
+ "27898 2.0 No 0 \n",
360
+ "27899 5.0 No 1 \n",
361
+ "27900 3.0 Yes 1 \n",
362
+ "\n",
363
+ "[27901 rows x 18 columns]"
364
+ ]
365
+ },
366
+ "execution_count": 95,
367
+ "metadata": {},
368
+ "output_type": "execute_result"
369
+ }
370
+ ],
371
+ "source": [
372
+ "df = pd.read_csv(r\"C:\\Users\\pavilion\\Desktop\\INNOMATICS\\Student_Depresion_Prediction(Machine_Learning)\\student_depression_dataset.csv\")\n",
373
+ "df"
374
+ ]
375
+ },
376
+ {
377
+ "cell_type": "code",
378
+ "execution_count": 143,
379
+ "id": "4e969e45-7a97-4c43-986a-7c34ce18682e",
380
+ "metadata": {},
381
+ "outputs": [
382
+ {
383
+ "data": {
384
+ "text/html": [
385
+ "<div>\n",
386
+ "<style scoped>\n",
387
+ " .dataframe tbody tr th:only-of-type {\n",
388
+ " vertical-align: middle;\n",
389
+ " }\n",
390
+ "\n",
391
+ " .dataframe tbody tr th {\n",
392
+ " vertical-align: top;\n",
393
+ " }\n",
394
+ "\n",
395
+ " .dataframe thead th {\n",
396
+ " text-align: right;\n",
397
+ " }\n",
398
+ "</style>\n",
399
+ "<table border=\"1\" class=\"dataframe\">\n",
400
+ " <thead>\n",
401
+ " <tr style=\"text-align: right;\">\n",
402
+ " <th></th>\n",
403
+ " <th>id</th>\n",
404
+ " <th>Gender</th>\n",
405
+ " <th>Age</th>\n",
406
+ " <th>City</th>\n",
407
+ " <th>Profession</th>\n",
408
+ " <th>Academic Pressure</th>\n",
409
+ " <th>Work Pressure</th>\n",
410
+ " <th>CGPA</th>\n",
411
+ " <th>Study Satisfaction</th>\n",
412
+ " <th>Job Satisfaction</th>\n",
413
+ " <th>Sleep Duration</th>\n",
414
+ " <th>Dietary Habits</th>\n",
415
+ " <th>Degree</th>\n",
416
+ " <th>Have you ever had suicidal thoughts ?</th>\n",
417
+ " <th>Work/Study Hours</th>\n",
418
+ " <th>Financial Stress</th>\n",
419
+ " <th>Family History of Mental Illness</th>\n",
420
+ " <th>Depression</th>\n",
421
+ " </tr>\n",
422
+ " </thead>\n",
423
+ " <tbody>\n",
424
+ " <tr>\n",
425
+ " <th>0</th>\n",
426
+ " <td>2</td>\n",
427
+ " <td>Male</td>\n",
428
+ " <td>33.0</td>\n",
429
+ " <td>Visakhapatnam</td>\n",
430
+ " <td>Student</td>\n",
431
+ " <td>5.0</td>\n",
432
+ " <td>0.0</td>\n",
433
+ " <td>8.97</td>\n",
434
+ " <td>2.0</td>\n",
435
+ " <td>0.0</td>\n",
436
+ " <td>'5-6 hours'</td>\n",
437
+ " <td>Healthy</td>\n",
438
+ " <td>B.Pharm</td>\n",
439
+ " <td>Yes</td>\n",
440
+ " <td>3.0</td>\n",
441
+ " <td>1.0</td>\n",
442
+ " <td>No</td>\n",
443
+ " <td>1</td>\n",
444
+ " </tr>\n",
445
+ " <tr>\n",
446
+ " <th>1</th>\n",
447
+ " <td>8</td>\n",
448
+ " <td>Female</td>\n",
449
+ " <td>24.0</td>\n",
450
+ " <td>Bangalore</td>\n",
451
+ " <td>Student</td>\n",
452
+ " <td>2.0</td>\n",
453
+ " <td>0.0</td>\n",
454
+ " <td>5.90</td>\n",
455
+ " <td>5.0</td>\n",
456
+ " <td>0.0</td>\n",
457
+ " <td>'5-6 hours'</td>\n",
458
+ " <td>Moderate</td>\n",
459
+ " <td>BSc</td>\n",
460
+ " <td>No</td>\n",
461
+ " <td>3.0</td>\n",
462
+ " <td>2.0</td>\n",
463
+ " <td>Yes</td>\n",
464
+ " <td>0</td>\n",
465
+ " </tr>\n",
466
+ " <tr>\n",
467
+ " <th>2</th>\n",
468
+ " <td>26</td>\n",
469
+ " <td>Male</td>\n",
470
+ " <td>31.0</td>\n",
471
+ " <td>Srinagar</td>\n",
472
+ " <td>Student</td>\n",
473
+ " <td>3.0</td>\n",
474
+ " <td>0.0</td>\n",
475
+ " <td>7.03</td>\n",
476
+ " <td>5.0</td>\n",
477
+ " <td>0.0</td>\n",
478
+ " <td>'Less than 5 hours'</td>\n",
479
+ " <td>Healthy</td>\n",
480
+ " <td>BA</td>\n",
481
+ " <td>No</td>\n",
482
+ " <td>9.0</td>\n",
483
+ " <td>1.0</td>\n",
484
+ " <td>Yes</td>\n",
485
+ " <td>0</td>\n",
486
+ " </tr>\n",
487
+ " <tr>\n",
488
+ " <th>3</th>\n",
489
+ " <td>30</td>\n",
490
+ " <td>Female</td>\n",
491
+ " <td>28.0</td>\n",
492
+ " <td>Varanasi</td>\n",
493
+ " <td>Student</td>\n",
494
+ " <td>3.0</td>\n",
495
+ " <td>0.0</td>\n",
496
+ " <td>5.59</td>\n",
497
+ " <td>2.0</td>\n",
498
+ " <td>0.0</td>\n",
499
+ " <td>'7-8 hours'</td>\n",
500
+ " <td>Moderate</td>\n",
501
+ " <td>BCA</td>\n",
502
+ " <td>Yes</td>\n",
503
+ " <td>4.0</td>\n",
504
+ " <td>5.0</td>\n",
505
+ " <td>Yes</td>\n",
506
+ " <td>1</td>\n",
507
+ " </tr>\n",
508
+ " <tr>\n",
509
+ " <th>4</th>\n",
510
+ " <td>32</td>\n",
511
+ " <td>Female</td>\n",
512
+ " <td>25.0</td>\n",
513
+ " <td>Jaipur</td>\n",
514
+ " <td>Student</td>\n",
515
+ " <td>4.0</td>\n",
516
+ " <td>0.0</td>\n",
517
+ " <td>8.13</td>\n",
518
+ " <td>3.0</td>\n",
519
+ " <td>0.0</td>\n",
520
+ " <td>'5-6 hours'</td>\n",
521
+ " <td>Moderate</td>\n",
522
+ " <td>M.Tech</td>\n",
523
+ " <td>Yes</td>\n",
524
+ " <td>1.0</td>\n",
525
+ " <td>1.0</td>\n",
526
+ " <td>No</td>\n",
527
+ " <td>0</td>\n",
528
+ " </tr>\n",
529
+ " <tr>\n",
530
+ " <th>...</th>\n",
531
+ " <td>...</td>\n",
532
+ " <td>...</td>\n",
533
+ " <td>...</td>\n",
534
+ " <td>...</td>\n",
535
+ " <td>...</td>\n",
536
+ " <td>...</td>\n",
537
+ " <td>...</td>\n",
538
+ " <td>...</td>\n",
539
+ " <td>...</td>\n",
540
+ " <td>...</td>\n",
541
+ " <td>...</td>\n",
542
+ " <td>...</td>\n",
543
+ " <td>...</td>\n",
544
+ " <td>...</td>\n",
545
+ " <td>...</td>\n",
546
+ " <td>...</td>\n",
547
+ " <td>...</td>\n",
548
+ " <td>...</td>\n",
549
+ " </tr>\n",
550
+ " <tr>\n",
551
+ " <th>27865</th>\n",
552
+ " <td>140685</td>\n",
553
+ " <td>Female</td>\n",
554
+ " <td>27.0</td>\n",
555
+ " <td>Surat</td>\n",
556
+ " <td>Student</td>\n",
557
+ " <td>5.0</td>\n",
558
+ " <td>0.0</td>\n",
559
+ " <td>5.75</td>\n",
560
+ " <td>5.0</td>\n",
561
+ " <td>0.0</td>\n",
562
+ " <td>'5-6 hours'</td>\n",
563
+ " <td>Unhealthy</td>\n",
564
+ " <td>'Class 12'</td>\n",
565
+ " <td>Yes</td>\n",
566
+ " <td>7.0</td>\n",
567
+ " <td>1.0</td>\n",
568
+ " <td>Yes</td>\n",
569
+ " <td>0</td>\n",
570
+ " </tr>\n",
571
+ " <tr>\n",
572
+ " <th>27866</th>\n",
573
+ " <td>140686</td>\n",
574
+ " <td>Male</td>\n",
575
+ " <td>27.0</td>\n",
576
+ " <td>Ludhiana</td>\n",
577
+ " <td>Student</td>\n",
578
+ " <td>2.0</td>\n",
579
+ " <td>0.0</td>\n",
580
+ " <td>9.40</td>\n",
581
+ " <td>3.0</td>\n",
582
+ " <td>0.0</td>\n",
583
+ " <td>'Less than 5 hours'</td>\n",
584
+ " <td>Healthy</td>\n",
585
+ " <td>MSc</td>\n",
586
+ " <td>No</td>\n",
587
+ " <td>0.0</td>\n",
588
+ " <td>3.0</td>\n",
589
+ " <td>Yes</td>\n",
590
+ " <td>0</td>\n",
591
+ " </tr>\n",
592
+ " <tr>\n",
593
+ " <th>27867</th>\n",
594
+ " <td>140689</td>\n",
595
+ " <td>Male</td>\n",
596
+ " <td>31.0</td>\n",
597
+ " <td>Faridabad</td>\n",
598
+ " <td>Student</td>\n",
599
+ " <td>3.0</td>\n",
600
+ " <td>0.0</td>\n",
601
+ " <td>6.61</td>\n",
602
+ " <td>4.0</td>\n",
603
+ " <td>0.0</td>\n",
604
+ " <td>'5-6 hours'</td>\n",
605
+ " <td>Unhealthy</td>\n",
606
+ " <td>MD</td>\n",
607
+ " <td>No</td>\n",
608
+ " <td>12.0</td>\n",
609
+ " <td>2.0</td>\n",
610
+ " <td>No</td>\n",
611
+ " <td>0</td>\n",
612
+ " </tr>\n",
613
+ " <tr>\n",
614
+ " <th>27868</th>\n",
615
+ " <td>140690</td>\n",
616
+ " <td>Female</td>\n",
617
+ " <td>18.0</td>\n",
618
+ " <td>Ludhiana</td>\n",
619
+ " <td>Student</td>\n",
620
+ " <td>5.0</td>\n",
621
+ " <td>0.0</td>\n",
622
+ " <td>6.88</td>\n",
623
+ " <td>2.0</td>\n",
624
+ " <td>0.0</td>\n",
625
+ " <td>'Less than 5 hours'</td>\n",
626
+ " <td>Healthy</td>\n",
627
+ " <td>'Class 12'</td>\n",
628
+ " <td>Yes</td>\n",
629
+ " <td>10.0</td>\n",
630
+ " <td>5.0</td>\n",
631
+ " <td>No</td>\n",
632
+ " <td>1</td>\n",
633
+ " </tr>\n",
634
+ " <tr>\n",
635
+ " <th>27869</th>\n",
636
+ " <td>140699</td>\n",
637
+ " <td>Male</td>\n",
638
+ " <td>27.0</td>\n",
639
+ " <td>Patna</td>\n",
640
+ " <td>Student</td>\n",
641
+ " <td>4.0</td>\n",
642
+ " <td>0.0</td>\n",
643
+ " <td>9.24</td>\n",
644
+ " <td>1.0</td>\n",
645
+ " <td>0.0</td>\n",
646
+ " <td>'Less than 5 hours'</td>\n",
647
+ " <td>Healthy</td>\n",
648
+ " <td>BCA</td>\n",
649
+ " <td>Yes</td>\n",
650
+ " <td>2.0</td>\n",
651
+ " <td>3.0</td>\n",
652
+ " <td>Yes</td>\n",
653
+ " <td>1</td>\n",
654
+ " </tr>\n",
655
+ " </tbody>\n",
656
+ "</table>\n",
657
+ "<p>27870 rows × 18 columns</p>\n",
658
+ "</div>"
659
+ ],
660
+ "text/plain": [
661
+ " id Gender Age City Profession Academic Pressure \\\n",
662
+ "0 2 Male 33.0 Visakhapatnam Student 5.0 \n",
663
+ "1 8 Female 24.0 Bangalore Student 2.0 \n",
664
+ "2 26 Male 31.0 Srinagar Student 3.0 \n",
665
+ "3 30 Female 28.0 Varanasi Student 3.0 \n",
666
+ "4 32 Female 25.0 Jaipur Student 4.0 \n",
667
+ "... ... ... ... ... ... ... \n",
668
+ "27865 140685 Female 27.0 Surat Student 5.0 \n",
669
+ "27866 140686 Male 27.0 Ludhiana Student 2.0 \n",
670
+ "27867 140689 Male 31.0 Faridabad Student 3.0 \n",
671
+ "27868 140690 Female 18.0 Ludhiana Student 5.0 \n",
672
+ "27869 140699 Male 27.0 Patna Student 4.0 \n",
673
+ "\n",
674
+ " Work Pressure CGPA Study Satisfaction Job Satisfaction \\\n",
675
+ "0 0.0 8.97 2.0 0.0 \n",
676
+ "1 0.0 5.90 5.0 0.0 \n",
677
+ "2 0.0 7.03 5.0 0.0 \n",
678
+ "3 0.0 5.59 2.0 0.0 \n",
679
+ "4 0.0 8.13 3.0 0.0 \n",
680
+ "... ... ... ... ... \n",
681
+ "27865 0.0 5.75 5.0 0.0 \n",
682
+ "27866 0.0 9.40 3.0 0.0 \n",
683
+ "27867 0.0 6.61 4.0 0.0 \n",
684
+ "27868 0.0 6.88 2.0 0.0 \n",
685
+ "27869 0.0 9.24 1.0 0.0 \n",
686
+ "\n",
687
+ " Sleep Duration Dietary Habits Degree \\\n",
688
+ "0 '5-6 hours' Healthy B.Pharm \n",
689
+ "1 '5-6 hours' Moderate BSc \n",
690
+ "2 'Less than 5 hours' Healthy BA \n",
691
+ "3 '7-8 hours' Moderate BCA \n",
692
+ "4 '5-6 hours' Moderate M.Tech \n",
693
+ "... ... ... ... \n",
694
+ "27865 '5-6 hours' Unhealthy 'Class 12' \n",
695
+ "27866 'Less than 5 hours' Healthy MSc \n",
696
+ "27867 '5-6 hours' Unhealthy MD \n",
697
+ "27868 'Less than 5 hours' Healthy 'Class 12' \n",
698
+ "27869 'Less than 5 hours' Healthy BCA \n",
699
+ "\n",
700
+ " Have you ever had suicidal thoughts ? Work/Study Hours \\\n",
701
+ "0 Yes 3.0 \n",
702
+ "1 No 3.0 \n",
703
+ "2 No 9.0 \n",
704
+ "3 Yes 4.0 \n",
705
+ "4 Yes 1.0 \n",
706
+ "... ... ... \n",
707
+ "27865 Yes 7.0 \n",
708
+ "27866 No 0.0 \n",
709
+ "27867 No 12.0 \n",
710
+ "27868 Yes 10.0 \n",
711
+ "27869 Yes 2.0 \n",
712
+ "\n",
713
+ " Financial Stress Family History of Mental Illness Depression \n",
714
+ "0 1.0 No 1 \n",
715
+ "1 2.0 Yes 0 \n",
716
+ "2 1.0 Yes 0 \n",
717
+ "3 5.0 Yes 1 \n",
718
+ "4 1.0 No 0 \n",
719
+ "... ... ... ... \n",
720
+ "27865 1.0 Yes 0 \n",
721
+ "27866 3.0 Yes 0 \n",
722
+ "27867 2.0 No 0 \n",
723
+ "27868 5.0 No 1 \n",
724
+ "27869 3.0 Yes 1 \n",
725
+ "\n",
726
+ "[27870 rows x 18 columns]"
727
+ ]
728
+ },
729
+ "execution_count": 143,
730
+ "metadata": {},
731
+ "output_type": "execute_result"
732
+ }
733
+ ],
734
+ "source": [
735
+ "df1 = df.loc[df[\"Profession\"]==\"Student\"].reset_index(drop=True)\n",
736
+ "df1"
737
+ ]
738
+ },
739
+ {
740
+ "cell_type": "code",
741
+ "execution_count": 145,
742
+ "id": "f51d9817-8b49-40d0-821c-35baf4cdfb88",
743
+ "metadata": {},
744
+ "outputs": [],
745
+ "source": [
746
+ "df1.to_csv(\"Student_Depresion_Prediction.csv\")"
747
+ ]
748
+ },
749
+ {
750
+ "cell_type": "code",
751
+ "execution_count": 152,
752
+ "id": "1cead9da-c26b-408f-9fbc-bb2fb93c92a2",
753
+ "metadata": {},
754
+ "outputs": [],
755
+ "source": [
756
+ "df1.drop(columns=[\"id\", \"City\", \"Job Satisfaction\", \"Work Pressure\"], inplace=True)"
757
+ ]
758
+ },
759
+ {
760
+ "cell_type": "code",
761
+ "execution_count": 156,
762
+ "id": "0c8970ce-68a1-492f-9b0c-fae93e8dd4fe",
763
+ "metadata": {},
764
+ "outputs": [
765
+ {
766
+ "data": {
767
+ "text/html": [
768
+ "<div>\n",
769
+ "<style scoped>\n",
770
+ " .dataframe tbody tr th:only-of-type {\n",
771
+ " vertical-align: middle;\n",
772
+ " }\n",
773
+ "\n",
774
+ " .dataframe tbody tr th {\n",
775
+ " vertical-align: top;\n",
776
+ " }\n",
777
+ "\n",
778
+ " .dataframe thead th {\n",
779
+ " text-align: right;\n",
780
+ " }\n",
781
+ "</style>\n",
782
+ "<table border=\"1\" class=\"dataframe\">\n",
783
+ " <thead>\n",
784
+ " <tr style=\"text-align: right;\">\n",
785
+ " <th></th>\n",
786
+ " <th>Gender</th>\n",
787
+ " <th>Age</th>\n",
788
+ " <th>Profession</th>\n",
789
+ " <th>Academic Pressure</th>\n",
790
+ " <th>CGPA</th>\n",
791
+ " <th>Study Satisfaction</th>\n",
792
+ " <th>Sleep Duration</th>\n",
793
+ " <th>Dietary Habits</th>\n",
794
+ " <th>Degree</th>\n",
795
+ " <th>Have you ever had suicidal thoughts ?</th>\n",
796
+ " <th>Work/Study Hours</th>\n",
797
+ " <th>Financial Stress</th>\n",
798
+ " <th>Family History of Mental Illness</th>\n",
799
+ " <th>Depression</th>\n",
800
+ " </tr>\n",
801
+ " </thead>\n",
802
+ " <tbody>\n",
803
+ " <tr>\n",
804
+ " <th>0</th>\n",
805
+ " <td>Male</td>\n",
806
+ " <td>33.0</td>\n",
807
+ " <td>Student</td>\n",
808
+ " <td>5.0</td>\n",
809
+ " <td>8.97</td>\n",
810
+ " <td>2.0</td>\n",
811
+ " <td>'5-6 hours'</td>\n",
812
+ " <td>Healthy</td>\n",
813
+ " <td>B.Pharm</td>\n",
814
+ " <td>Yes</td>\n",
815
+ " <td>3.0</td>\n",
816
+ " <td>1.0</td>\n",
817
+ " <td>No</td>\n",
818
+ " <td>1</td>\n",
819
+ " </tr>\n",
820
+ " <tr>\n",
821
+ " <th>1</th>\n",
822
+ " <td>Female</td>\n",
823
+ " <td>24.0</td>\n",
824
+ " <td>Student</td>\n",
825
+ " <td>2.0</td>\n",
826
+ " <td>5.90</td>\n",
827
+ " <td>5.0</td>\n",
828
+ " <td>'5-6 hours'</td>\n",
829
+ " <td>Moderate</td>\n",
830
+ " <td>BSc</td>\n",
831
+ " <td>No</td>\n",
832
+ " <td>3.0</td>\n",
833
+ " <td>2.0</td>\n",
834
+ " <td>Yes</td>\n",
835
+ " <td>0</td>\n",
836
+ " </tr>\n",
837
+ " <tr>\n",
838
+ " <th>2</th>\n",
839
+ " <td>Male</td>\n",
840
+ " <td>31.0</td>\n",
841
+ " <td>Student</td>\n",
842
+ " <td>3.0</td>\n",
843
+ " <td>7.03</td>\n",
844
+ " <td>5.0</td>\n",
845
+ " <td>'Less than 5 hours'</td>\n",
846
+ " <td>Healthy</td>\n",
847
+ " <td>BA</td>\n",
848
+ " <td>No</td>\n",
849
+ " <td>9.0</td>\n",
850
+ " <td>1.0</td>\n",
851
+ " <td>Yes</td>\n",
852
+ " <td>0</td>\n",
853
+ " </tr>\n",
854
+ " <tr>\n",
855
+ " <th>3</th>\n",
856
+ " <td>Female</td>\n",
857
+ " <td>28.0</td>\n",
858
+ " <td>Student</td>\n",
859
+ " <td>3.0</td>\n",
860
+ " <td>5.59</td>\n",
861
+ " <td>2.0</td>\n",
862
+ " <td>'7-8 hours'</td>\n",
863
+ " <td>Moderate</td>\n",
864
+ " <td>BCA</td>\n",
865
+ " <td>Yes</td>\n",
866
+ " <td>4.0</td>\n",
867
+ " <td>5.0</td>\n",
868
+ " <td>Yes</td>\n",
869
+ " <td>1</td>\n",
870
+ " </tr>\n",
871
+ " <tr>\n",
872
+ " <th>4</th>\n",
873
+ " <td>Female</td>\n",
874
+ " <td>25.0</td>\n",
875
+ " <td>Student</td>\n",
876
+ " <td>4.0</td>\n",
877
+ " <td>8.13</td>\n",
878
+ " <td>3.0</td>\n",
879
+ " <td>'5-6 hours'</td>\n",
880
+ " <td>Moderate</td>\n",
881
+ " <td>M.Tech</td>\n",
882
+ " <td>Yes</td>\n",
883
+ " <td>1.0</td>\n",
884
+ " <td>1.0</td>\n",
885
+ " <td>No</td>\n",
886
+ " <td>0</td>\n",
887
+ " </tr>\n",
888
+ " <tr>\n",
889
+ " <th>...</th>\n",
890
+ " <td>...</td>\n",
891
+ " <td>...</td>\n",
892
+ " <td>...</td>\n",
893
+ " <td>...</td>\n",
894
+ " <td>...</td>\n",
895
+ " <td>...</td>\n",
896
+ " <td>...</td>\n",
897
+ " <td>...</td>\n",
898
+ " <td>...</td>\n",
899
+ " <td>...</td>\n",
900
+ " <td>...</td>\n",
901
+ " <td>...</td>\n",
902
+ " <td>...</td>\n",
903
+ " <td>...</td>\n",
904
+ " </tr>\n",
905
+ " <tr>\n",
906
+ " <th>27865</th>\n",
907
+ " <td>Female</td>\n",
908
+ " <td>27.0</td>\n",
909
+ " <td>Student</td>\n",
910
+ " <td>5.0</td>\n",
911
+ " <td>5.75</td>\n",
912
+ " <td>5.0</td>\n",
913
+ " <td>'5-6 hours'</td>\n",
914
+ " <td>Unhealthy</td>\n",
915
+ " <td>'Class 12'</td>\n",
916
+ " <td>Yes</td>\n",
917
+ " <td>7.0</td>\n",
918
+ " <td>1.0</td>\n",
919
+ " <td>Yes</td>\n",
920
+ " <td>0</td>\n",
921
+ " </tr>\n",
922
+ " <tr>\n",
923
+ " <th>27866</th>\n",
924
+ " <td>Male</td>\n",
925
+ " <td>27.0</td>\n",
926
+ " <td>Student</td>\n",
927
+ " <td>2.0</td>\n",
928
+ " <td>9.40</td>\n",
929
+ " <td>3.0</td>\n",
930
+ " <td>'Less than 5 hours'</td>\n",
931
+ " <td>Healthy</td>\n",
932
+ " <td>MSc</td>\n",
933
+ " <td>No</td>\n",
934
+ " <td>0.0</td>\n",
935
+ " <td>3.0</td>\n",
936
+ " <td>Yes</td>\n",
937
+ " <td>0</td>\n",
938
+ " </tr>\n",
939
+ " <tr>\n",
940
+ " <th>27867</th>\n",
941
+ " <td>Male</td>\n",
942
+ " <td>31.0</td>\n",
943
+ " <td>Student</td>\n",
944
+ " <td>3.0</td>\n",
945
+ " <td>6.61</td>\n",
946
+ " <td>4.0</td>\n",
947
+ " <td>'5-6 hours'</td>\n",
948
+ " <td>Unhealthy</td>\n",
949
+ " <td>MD</td>\n",
950
+ " <td>No</td>\n",
951
+ " <td>12.0</td>\n",
952
+ " <td>2.0</td>\n",
953
+ " <td>No</td>\n",
954
+ " <td>0</td>\n",
955
+ " </tr>\n",
956
+ " <tr>\n",
957
+ " <th>27868</th>\n",
958
+ " <td>Female</td>\n",
959
+ " <td>18.0</td>\n",
960
+ " <td>Student</td>\n",
961
+ " <td>5.0</td>\n",
962
+ " <td>6.88</td>\n",
963
+ " <td>2.0</td>\n",
964
+ " <td>'Less than 5 hours'</td>\n",
965
+ " <td>Healthy</td>\n",
966
+ " <td>'Class 12'</td>\n",
967
+ " <td>Yes</td>\n",
968
+ " <td>10.0</td>\n",
969
+ " <td>5.0</td>\n",
970
+ " <td>No</td>\n",
971
+ " <td>1</td>\n",
972
+ " </tr>\n",
973
+ " <tr>\n",
974
+ " <th>27869</th>\n",
975
+ " <td>Male</td>\n",
976
+ " <td>27.0</td>\n",
977
+ " <td>Student</td>\n",
978
+ " <td>4.0</td>\n",
979
+ " <td>9.24</td>\n",
980
+ " <td>1.0</td>\n",
981
+ " <td>'Less than 5 hours'</td>\n",
982
+ " <td>Healthy</td>\n",
983
+ " <td>BCA</td>\n",
984
+ " <td>Yes</td>\n",
985
+ " <td>2.0</td>\n",
986
+ " <td>3.0</td>\n",
987
+ " <td>Yes</td>\n",
988
+ " <td>1</td>\n",
989
+ " </tr>\n",
990
+ " </tbody>\n",
991
+ "</table>\n",
992
+ "<p>27870 rows × 14 columns</p>\n",
993
+ "</div>"
994
+ ],
995
+ "text/plain": [
996
+ " Gender Age Profession Academic Pressure CGPA Study Satisfaction \\\n",
997
+ "0 Male 33.0 Student 5.0 8.97 2.0 \n",
998
+ "1 Female 24.0 Student 2.0 5.90 5.0 \n",
999
+ "2 Male 31.0 Student 3.0 7.03 5.0 \n",
1000
+ "3 Female 28.0 Student 3.0 5.59 2.0 \n",
1001
+ "4 Female 25.0 Student 4.0 8.13 3.0 \n",
1002
+ "... ... ... ... ... ... ... \n",
1003
+ "27865 Female 27.0 Student 5.0 5.75 5.0 \n",
1004
+ "27866 Male 27.0 Student 2.0 9.40 3.0 \n",
1005
+ "27867 Male 31.0 Student 3.0 6.61 4.0 \n",
1006
+ "27868 Female 18.0 Student 5.0 6.88 2.0 \n",
1007
+ "27869 Male 27.0 Student 4.0 9.24 1.0 \n",
1008
+ "\n",
1009
+ " Sleep Duration Dietary Habits Degree \\\n",
1010
+ "0 '5-6 hours' Healthy B.Pharm \n",
1011
+ "1 '5-6 hours' Moderate BSc \n",
1012
+ "2 'Less than 5 hours' Healthy BA \n",
1013
+ "3 '7-8 hours' Moderate BCA \n",
1014
+ "4 '5-6 hours' Moderate M.Tech \n",
1015
+ "... ... ... ... \n",
1016
+ "27865 '5-6 hours' Unhealthy 'Class 12' \n",
1017
+ "27866 'Less than 5 hours' Healthy MSc \n",
1018
+ "27867 '5-6 hours' Unhealthy MD \n",
1019
+ "27868 'Less than 5 hours' Healthy 'Class 12' \n",
1020
+ "27869 'Less than 5 hours' Healthy BCA \n",
1021
+ "\n",
1022
+ " Have you ever had suicidal thoughts ? Work/Study Hours \\\n",
1023
+ "0 Yes 3.0 \n",
1024
+ "1 No 3.0 \n",
1025
+ "2 No 9.0 \n",
1026
+ "3 Yes 4.0 \n",
1027
+ "4 Yes 1.0 \n",
1028
+ "... ... ... \n",
1029
+ "27865 Yes 7.0 \n",
1030
+ "27866 No 0.0 \n",
1031
+ "27867 No 12.0 \n",
1032
+ "27868 Yes 10.0 \n",
1033
+ "27869 Yes 2.0 \n",
1034
+ "\n",
1035
+ " Financial Stress Family History of Mental Illness Depression \n",
1036
+ "0 1.0 No 1 \n",
1037
+ "1 2.0 Yes 0 \n",
1038
+ "2 1.0 Yes 0 \n",
1039
+ "3 5.0 Yes 1 \n",
1040
+ "4 1.0 No 0 \n",
1041
+ "... ... ... ... \n",
1042
+ "27865 1.0 Yes 0 \n",
1043
+ "27866 3.0 Yes 0 \n",
1044
+ "27867 2.0 No 0 \n",
1045
+ "27868 5.0 No 1 \n",
1046
+ "27869 3.0 Yes 1 \n",
1047
+ "\n",
1048
+ "[27870 rows x 14 columns]"
1049
+ ]
1050
+ },
1051
+ "execution_count": 156,
1052
+ "metadata": {},
1053
+ "output_type": "execute_result"
1054
+ }
1055
+ ],
1056
+ "source": [
1057
+ "df1"
1058
+ ]
1059
+ },
1060
+ {
1061
+ "cell_type": "code",
1062
+ "execution_count": 158,
1063
+ "id": "8b272d83-19d8-4f89-97e9-d3d02e4d55fd",
1064
+ "metadata": {},
1065
+ "outputs": [
1066
+ {
1067
+ "data": {
1068
+ "text/plain": [
1069
+ "(27870, 14)"
1070
+ ]
1071
+ },
1072
+ "execution_count": 158,
1073
+ "metadata": {},
1074
+ "output_type": "execute_result"
1075
+ }
1076
+ ],
1077
+ "source": [
1078
+ "df1.shape"
1079
+ ]
1080
+ },
1081
+ {
1082
+ "cell_type": "code",
1083
+ "execution_count": 162,
1084
+ "id": "fbf8de3f-dd98-4613-b3f1-5f2a48e0c7e1",
1085
+ "metadata": {},
1086
+ "outputs": [
1087
+ {
1088
+ "name": "stdout",
1089
+ "output_type": "stream",
1090
+ "text": [
1091
+ "<class 'pandas.core.frame.DataFrame'>\n",
1092
+ "RangeIndex: 27870 entries, 0 to 27869\n",
1093
+ "Data columns (total 14 columns):\n",
1094
+ " # Column Non-Null Count Dtype \n",
1095
+ "--- ------ -------------- ----- \n",
1096
+ " 0 Gender 27870 non-null object \n",
1097
+ " 1 Age 27870 non-null float64\n",
1098
+ " 2 Profession 27870 non-null object \n",
1099
+ " 3 Academic Pressure 27870 non-null float64\n",
1100
+ " 4 CGPA 27870 non-null float64\n",
1101
+ " 5 Study Satisfaction 27870 non-null float64\n",
1102
+ " 6 Sleep Duration 27870 non-null object \n",
1103
+ " 7 Dietary Habits 27870 non-null object \n",
1104
+ " 8 Degree 27870 non-null object \n",
1105
+ " 9 Have you ever had suicidal thoughts ? 27870 non-null object \n",
1106
+ " 10 Work/Study Hours 27870 non-null float64\n",
1107
+ " 11 Financial Stress 27870 non-null object \n",
1108
+ " 12 Family History of Mental Illness 27870 non-null object \n",
1109
+ " 13 Depression 27870 non-null int64 \n",
1110
+ "dtypes: float64(5), int64(1), object(8)\n",
1111
+ "memory usage: 3.0+ MB\n"
1112
+ ]
1113
+ }
1114
+ ],
1115
+ "source": [
1116
+ "df1.info()"
1117
+ ]
1118
+ },
1119
+ {
1120
+ "cell_type": "code",
1121
+ "execution_count": 166,
1122
+ "id": "0a47a407-c42c-485d-8ecf-002ff4eb1d33",
1123
+ "metadata": {},
1124
+ "outputs": [
1125
+ {
1126
+ "data": {
1127
+ "text/html": [
1128
+ "<div>\n",
1129
+ "<style scoped>\n",
1130
+ " .dataframe tbody tr th:only-of-type {\n",
1131
+ " vertical-align: middle;\n",
1132
+ " }\n",
1133
+ "\n",
1134
+ " .dataframe tbody tr th {\n",
1135
+ " vertical-align: top;\n",
1136
+ " }\n",
1137
+ "\n",
1138
+ " .dataframe thead th {\n",
1139
+ " text-align: right;\n",
1140
+ " }\n",
1141
+ "</style>\n",
1142
+ "<table border=\"1\" class=\"dataframe\">\n",
1143
+ " <thead>\n",
1144
+ " <tr style=\"text-align: right;\">\n",
1145
+ " <th></th>\n",
1146
+ " <th>Gender</th>\n",
1147
+ " <th>Age</th>\n",
1148
+ " <th>Profession</th>\n",
1149
+ " <th>Academic Pressure</th>\n",
1150
+ " <th>CGPA</th>\n",
1151
+ " <th>Study Satisfaction</th>\n",
1152
+ " <th>Sleep Duration</th>\n",
1153
+ " <th>Dietary Habits</th>\n",
1154
+ " <th>Degree</th>\n",
1155
+ " <th>Have you ever had suicidal thoughts ?</th>\n",
1156
+ " <th>Work/Study Hours</th>\n",
1157
+ " <th>Financial Stress</th>\n",
1158
+ " <th>Family History of Mental Illness</th>\n",
1159
+ " <th>Depression</th>\n",
1160
+ " </tr>\n",
1161
+ " </thead>\n",
1162
+ " <tbody>\n",
1163
+ " <tr>\n",
1164
+ " <th>0</th>\n",
1165
+ " <td>Male</td>\n",
1166
+ " <td>33.0</td>\n",
1167
+ " <td>Student</td>\n",
1168
+ " <td>5.0</td>\n",
1169
+ " <td>8.97</td>\n",
1170
+ " <td>2.0</td>\n",
1171
+ " <td>'5-6 hours'</td>\n",
1172
+ " <td>Healthy</td>\n",
1173
+ " <td>B.Pharm</td>\n",
1174
+ " <td>Yes</td>\n",
1175
+ " <td>3.0</td>\n",
1176
+ " <td>1.0</td>\n",
1177
+ " <td>No</td>\n",
1178
+ " <td>1</td>\n",
1179
+ " </tr>\n",
1180
+ " <tr>\n",
1181
+ " <th>1</th>\n",
1182
+ " <td>Female</td>\n",
1183
+ " <td>24.0</td>\n",
1184
+ " <td>Student</td>\n",
1185
+ " <td>2.0</td>\n",
1186
+ " <td>5.90</td>\n",
1187
+ " <td>5.0</td>\n",
1188
+ " <td>'5-6 hours'</td>\n",
1189
+ " <td>Moderate</td>\n",
1190
+ " <td>BSc</td>\n",
1191
+ " <td>No</td>\n",
1192
+ " <td>3.0</td>\n",
1193
+ " <td>2.0</td>\n",
1194
+ " <td>Yes</td>\n",
1195
+ " <td>0</td>\n",
1196
+ " </tr>\n",
1197
+ " <tr>\n",
1198
+ " <th>2</th>\n",
1199
+ " <td>Male</td>\n",
1200
+ " <td>31.0</td>\n",
1201
+ " <td>Student</td>\n",
1202
+ " <td>3.0</td>\n",
1203
+ " <td>7.03</td>\n",
1204
+ " <td>5.0</td>\n",
1205
+ " <td>'Less than 5 hours'</td>\n",
1206
+ " <td>Healthy</td>\n",
1207
+ " <td>BA</td>\n",
1208
+ " <td>No</td>\n",
1209
+ " <td>9.0</td>\n",
1210
+ " <td>1.0</td>\n",
1211
+ " <td>Yes</td>\n",
1212
+ " <td>0</td>\n",
1213
+ " </tr>\n",
1214
+ " <tr>\n",
1215
+ " <th>3</th>\n",
1216
+ " <td>Female</td>\n",
1217
+ " <td>28.0</td>\n",
1218
+ " <td>Student</td>\n",
1219
+ " <td>3.0</td>\n",
1220
+ " <td>5.59</td>\n",
1221
+ " <td>2.0</td>\n",
1222
+ " <td>'7-8 hours'</td>\n",
1223
+ " <td>Moderate</td>\n",
1224
+ " <td>BCA</td>\n",
1225
+ " <td>Yes</td>\n",
1226
+ " <td>4.0</td>\n",
1227
+ " <td>5.0</td>\n",
1228
+ " <td>Yes</td>\n",
1229
+ " <td>1</td>\n",
1230
+ " </tr>\n",
1231
+ " <tr>\n",
1232
+ " <th>4</th>\n",
1233
+ " <td>Female</td>\n",
1234
+ " <td>25.0</td>\n",
1235
+ " <td>Student</td>\n",
1236
+ " <td>4.0</td>\n",
1237
+ " <td>8.13</td>\n",
1238
+ " <td>3.0</td>\n",
1239
+ " <td>'5-6 hours'</td>\n",
1240
+ " <td>Moderate</td>\n",
1241
+ " <td>M.Tech</td>\n",
1242
+ " <td>Yes</td>\n",
1243
+ " <td>1.0</td>\n",
1244
+ " <td>1.0</td>\n",
1245
+ " <td>No</td>\n",
1246
+ " <td>0</td>\n",
1247
+ " </tr>\n",
1248
+ " <tr>\n",
1249
+ " <th>...</th>\n",
1250
+ " <td>...</td>\n",
1251
+ " <td>...</td>\n",
1252
+ " <td>...</td>\n",
1253
+ " <td>...</td>\n",
1254
+ " <td>...</td>\n",
1255
+ " <td>...</td>\n",
1256
+ " <td>...</td>\n",
1257
+ " <td>...</td>\n",
1258
+ " <td>...</td>\n",
1259
+ " <td>...</td>\n",
1260
+ " <td>...</td>\n",
1261
+ " <td>...</td>\n",
1262
+ " <td>...</td>\n",
1263
+ " <td>...</td>\n",
1264
+ " </tr>\n",
1265
+ " <tr>\n",
1266
+ " <th>27865</th>\n",
1267
+ " <td>Female</td>\n",
1268
+ " <td>27.0</td>\n",
1269
+ " <td>Student</td>\n",
1270
+ " <td>5.0</td>\n",
1271
+ " <td>5.75</td>\n",
1272
+ " <td>5.0</td>\n",
1273
+ " <td>'5-6 hours'</td>\n",
1274
+ " <td>Unhealthy</td>\n",
1275
+ " <td>'Class 12'</td>\n",
1276
+ " <td>Yes</td>\n",
1277
+ " <td>7.0</td>\n",
1278
+ " <td>1.0</td>\n",
1279
+ " <td>Yes</td>\n",
1280
+ " <td>0</td>\n",
1281
+ " </tr>\n",
1282
+ " <tr>\n",
1283
+ " <th>27866</th>\n",
1284
+ " <td>Male</td>\n",
1285
+ " <td>27.0</td>\n",
1286
+ " <td>Student</td>\n",
1287
+ " <td>2.0</td>\n",
1288
+ " <td>9.40</td>\n",
1289
+ " <td>3.0</td>\n",
1290
+ " <td>'Less than 5 hours'</td>\n",
1291
+ " <td>Healthy</td>\n",
1292
+ " <td>MSc</td>\n",
1293
+ " <td>No</td>\n",
1294
+ " <td>0.0</td>\n",
1295
+ " <td>3.0</td>\n",
1296
+ " <td>Yes</td>\n",
1297
+ " <td>0</td>\n",
1298
+ " </tr>\n",
1299
+ " <tr>\n",
1300
+ " <th>27867</th>\n",
1301
+ " <td>Male</td>\n",
1302
+ " <td>31.0</td>\n",
1303
+ " <td>Student</td>\n",
1304
+ " <td>3.0</td>\n",
1305
+ " <td>6.61</td>\n",
1306
+ " <td>4.0</td>\n",
1307
+ " <td>'5-6 hours'</td>\n",
1308
+ " <td>Unhealthy</td>\n",
1309
+ " <td>MD</td>\n",
1310
+ " <td>No</td>\n",
1311
+ " <td>12.0</td>\n",
1312
+ " <td>2.0</td>\n",
1313
+ " <td>No</td>\n",
1314
+ " <td>0</td>\n",
1315
+ " </tr>\n",
1316
+ " <tr>\n",
1317
+ " <th>27868</th>\n",
1318
+ " <td>Female</td>\n",
1319
+ " <td>18.0</td>\n",
1320
+ " <td>Student</td>\n",
1321
+ " <td>5.0</td>\n",
1322
+ " <td>6.88</td>\n",
1323
+ " <td>2.0</td>\n",
1324
+ " <td>'Less than 5 hours'</td>\n",
1325
+ " <td>Healthy</td>\n",
1326
+ " <td>'Class 12'</td>\n",
1327
+ " <td>Yes</td>\n",
1328
+ " <td>10.0</td>\n",
1329
+ " <td>5.0</td>\n",
1330
+ " <td>No</td>\n",
1331
+ " <td>1</td>\n",
1332
+ " </tr>\n",
1333
+ " <tr>\n",
1334
+ " <th>27869</th>\n",
1335
+ " <td>Male</td>\n",
1336
+ " <td>27.0</td>\n",
1337
+ " <td>Student</td>\n",
1338
+ " <td>4.0</td>\n",
1339
+ " <td>9.24</td>\n",
1340
+ " <td>1.0</td>\n",
1341
+ " <td>'Less than 5 hours'</td>\n",
1342
+ " <td>Healthy</td>\n",
1343
+ " <td>BCA</td>\n",
1344
+ " <td>Yes</td>\n",
1345
+ " <td>2.0</td>\n",
1346
+ " <td>3.0</td>\n",
1347
+ " <td>Yes</td>\n",
1348
+ " <td>1</td>\n",
1349
+ " </tr>\n",
1350
+ " </tbody>\n",
1351
+ "</table>\n",
1352
+ "<p>27870 rows × 14 columns</p>\n",
1353
+ "</div>"
1354
+ ],
1355
+ "text/plain": [
1356
+ " Gender Age Profession Academic Pressure CGPA Study Satisfaction \\\n",
1357
+ "0 Male 33.0 Student 5.0 8.97 2.0 \n",
1358
+ "1 Female 24.0 Student 2.0 5.90 5.0 \n",
1359
+ "2 Male 31.0 Student 3.0 7.03 5.0 \n",
1360
+ "3 Female 28.0 Student 3.0 5.59 2.0 \n",
1361
+ "4 Female 25.0 Student 4.0 8.13 3.0 \n",
1362
+ "... ... ... ... ... ... ... \n",
1363
+ "27865 Female 27.0 Student 5.0 5.75 5.0 \n",
1364
+ "27866 Male 27.0 Student 2.0 9.40 3.0 \n",
1365
+ "27867 Male 31.0 Student 3.0 6.61 4.0 \n",
1366
+ "27868 Female 18.0 Student 5.0 6.88 2.0 \n",
1367
+ "27869 Male 27.0 Student 4.0 9.24 1.0 \n",
1368
+ "\n",
1369
+ " Sleep Duration Dietary Habits Degree \\\n",
1370
+ "0 '5-6 hours' Healthy B.Pharm \n",
1371
+ "1 '5-6 hours' Moderate BSc \n",
1372
+ "2 'Less than 5 hours' Healthy BA \n",
1373
+ "3 '7-8 hours' Moderate BCA \n",
1374
+ "4 '5-6 hours' Moderate M.Tech \n",
1375
+ "... ... ... ... \n",
1376
+ "27865 '5-6 hours' Unhealthy 'Class 12' \n",
1377
+ "27866 'Less than 5 hours' Healthy MSc \n",
1378
+ "27867 '5-6 hours' Unhealthy MD \n",
1379
+ "27868 'Less than 5 hours' Healthy 'Class 12' \n",
1380
+ "27869 'Less than 5 hours' Healthy BCA \n",
1381
+ "\n",
1382
+ " Have you ever had suicidal thoughts ? Work/Study Hours \\\n",
1383
+ "0 Yes 3.0 \n",
1384
+ "1 No 3.0 \n",
1385
+ "2 No 9.0 \n",
1386
+ "3 Yes 4.0 \n",
1387
+ "4 Yes 1.0 \n",
1388
+ "... ... ... \n",
1389
+ "27865 Yes 7.0 \n",
1390
+ "27866 No 0.0 \n",
1391
+ "27867 No 12.0 \n",
1392
+ "27868 Yes 10.0 \n",
1393
+ "27869 Yes 2.0 \n",
1394
+ "\n",
1395
+ " Financial Stress Family History of Mental Illness Depression \n",
1396
+ "0 1.0 No 1 \n",
1397
+ "1 2.0 Yes 0 \n",
1398
+ "2 1.0 Yes 0 \n",
1399
+ "3 5.0 Yes 1 \n",
1400
+ "4 1.0 No 0 \n",
1401
+ "... ... ... ... \n",
1402
+ "27865 1.0 Yes 0 \n",
1403
+ "27866 3.0 Yes 0 \n",
1404
+ "27867 2.0 No 0 \n",
1405
+ "27868 5.0 No 1 \n",
1406
+ "27869 3.0 Yes 1 \n",
1407
+ "\n",
1408
+ "[27870 rows x 14 columns]"
1409
+ ]
1410
+ },
1411
+ "execution_count": 166,
1412
+ "metadata": {},
1413
+ "output_type": "execute_result"
1414
+ }
1415
+ ],
1416
+ "source": [
1417
+ "duplicates = df1.duplicated().sum()\n",
1418
+ "df1 = df1.drop_duplicates().reset_index(drop=True)\n",
1419
+ "df1"
1420
+ ]
1421
+ },
1422
+ {
1423
+ "cell_type": "code",
1424
+ "execution_count": 183,
1425
+ "id": "82476012-5849-4a29-93c8-8bed0e24d7bf",
1426
+ "metadata": {},
1427
+ "outputs": [],
1428
+ "source": [
1429
+ "df1[\"Sleep Duration\"] = df1[\"Sleep Duration\"].str.replace(\"'\", \"\", regex=False)\n",
1430
+ "df1[\"Degree\"] = df1[\"Degree\"].str.replace(\"'\", \"\", regex=False)\n",
1431
+ "\n",
1432
+ "df1[\"Financial Stress\"] = df1[\"Financial Stress\"].replace(\"?\", df1[\"Financial Stress\"].mode()[0])"
1433
+ ]
1434
+ },
1435
+ {
1436
+ "cell_type": "code",
1437
+ "execution_count": 178,
1438
+ "id": "5c5f1d7d-7061-45dd-8a3c-86f9a090ab33",
1439
+ "metadata": {},
1440
+ "outputs": [
1441
+ {
1442
+ "data": {
1443
+ "text/html": [
1444
+ "<div>\n",
1445
+ "<style scoped>\n",
1446
+ " .dataframe tbody tr th:only-of-type {\n",
1447
+ " vertical-align: middle;\n",
1448
+ " }\n",
1449
+ "\n",
1450
+ " .dataframe tbody tr th {\n",
1451
+ " vertical-align: top;\n",
1452
+ " }\n",
1453
+ "\n",
1454
+ " .dataframe thead th {\n",
1455
+ " text-align: right;\n",
1456
+ " }\n",
1457
+ "</style>\n",
1458
+ "<table border=\"1\" class=\"dataframe\">\n",
1459
+ " <thead>\n",
1460
+ " <tr style=\"text-align: right;\">\n",
1461
+ " <th></th>\n",
1462
+ " <th>Gender</th>\n",
1463
+ " <th>Age</th>\n",
1464
+ " <th>Profession</th>\n",
1465
+ " <th>Academic Pressure</th>\n",
1466
+ " <th>CGPA</th>\n",
1467
+ " <th>Study Satisfaction</th>\n",
1468
+ " <th>Sleep Duration</th>\n",
1469
+ " <th>Dietary Habits</th>\n",
1470
+ " <th>Degree</th>\n",
1471
+ " <th>Have you ever had suicidal thoughts ?</th>\n",
1472
+ " <th>Work/Study Hours</th>\n",
1473
+ " <th>Financial Stress</th>\n",
1474
+ " <th>Family History of Mental Illness</th>\n",
1475
+ " <th>Depression</th>\n",
1476
+ " </tr>\n",
1477
+ " </thead>\n",
1478
+ " <tbody>\n",
1479
+ " <tr>\n",
1480
+ " <th>0</th>\n",
1481
+ " <td>Male</td>\n",
1482
+ " <td>33.0</td>\n",
1483
+ " <td>Student</td>\n",
1484
+ " <td>5.0</td>\n",
1485
+ " <td>8.97</td>\n",
1486
+ " <td>2.0</td>\n",
1487
+ " <td>5-6 hours</td>\n",
1488
+ " <td>Healthy</td>\n",
1489
+ " <td>B.Pharm</td>\n",
1490
+ " <td>Yes</td>\n",
1491
+ " <td>3.0</td>\n",
1492
+ " <td>1.0</td>\n",
1493
+ " <td>No</td>\n",
1494
+ " <td>1</td>\n",
1495
+ " </tr>\n",
1496
+ " <tr>\n",
1497
+ " <th>1</th>\n",
1498
+ " <td>Female</td>\n",
1499
+ " <td>24.0</td>\n",
1500
+ " <td>Student</td>\n",
1501
+ " <td>2.0</td>\n",
1502
+ " <td>5.90</td>\n",
1503
+ " <td>5.0</td>\n",
1504
+ " <td>5-6 hours</td>\n",
1505
+ " <td>Moderate</td>\n",
1506
+ " <td>BSc</td>\n",
1507
+ " <td>No</td>\n",
1508
+ " <td>3.0</td>\n",
1509
+ " <td>2.0</td>\n",
1510
+ " <td>Yes</td>\n",
1511
+ " <td>0</td>\n",
1512
+ " </tr>\n",
1513
+ " <tr>\n",
1514
+ " <th>2</th>\n",
1515
+ " <td>Male</td>\n",
1516
+ " <td>31.0</td>\n",
1517
+ " <td>Student</td>\n",
1518
+ " <td>3.0</td>\n",
1519
+ " <td>7.03</td>\n",
1520
+ " <td>5.0</td>\n",
1521
+ " <td>Less than 5 hours</td>\n",
1522
+ " <td>Healthy</td>\n",
1523
+ " <td>BA</td>\n",
1524
+ " <td>No</td>\n",
1525
+ " <td>9.0</td>\n",
1526
+ " <td>1.0</td>\n",
1527
+ " <td>Yes</td>\n",
1528
+ " <td>0</td>\n",
1529
+ " </tr>\n",
1530
+ " <tr>\n",
1531
+ " <th>3</th>\n",
1532
+ " <td>Female</td>\n",
1533
+ " <td>28.0</td>\n",
1534
+ " <td>Student</td>\n",
1535
+ " <td>3.0</td>\n",
1536
+ " <td>5.59</td>\n",
1537
+ " <td>2.0</td>\n",
1538
+ " <td>7-8 hours</td>\n",
1539
+ " <td>Moderate</td>\n",
1540
+ " <td>BCA</td>\n",
1541
+ " <td>Yes</td>\n",
1542
+ " <td>4.0</td>\n",
1543
+ " <td>5.0</td>\n",
1544
+ " <td>Yes</td>\n",
1545
+ " <td>1</td>\n",
1546
+ " </tr>\n",
1547
+ " <tr>\n",
1548
+ " <th>4</th>\n",
1549
+ " <td>Female</td>\n",
1550
+ " <td>25.0</td>\n",
1551
+ " <td>Student</td>\n",
1552
+ " <td>4.0</td>\n",
1553
+ " <td>8.13</td>\n",
1554
+ " <td>3.0</td>\n",
1555
+ " <td>5-6 hours</td>\n",
1556
+ " <td>Moderate</td>\n",
1557
+ " <td>M.Tech</td>\n",
1558
+ " <td>Yes</td>\n",
1559
+ " <td>1.0</td>\n",
1560
+ " <td>1.0</td>\n",
1561
+ " <td>No</td>\n",
1562
+ " <td>0</td>\n",
1563
+ " </tr>\n",
1564
+ " <tr>\n",
1565
+ " <th>...</th>\n",
1566
+ " <td>...</td>\n",
1567
+ " <td>...</td>\n",
1568
+ " <td>...</td>\n",
1569
+ " <td>...</td>\n",
1570
+ " <td>...</td>\n",
1571
+ " <td>...</td>\n",
1572
+ " <td>...</td>\n",
1573
+ " <td>...</td>\n",
1574
+ " <td>...</td>\n",
1575
+ " <td>...</td>\n",
1576
+ " <td>...</td>\n",
1577
+ " <td>...</td>\n",
1578
+ " <td>...</td>\n",
1579
+ " <td>...</td>\n",
1580
+ " </tr>\n",
1581
+ " <tr>\n",
1582
+ " <th>27865</th>\n",
1583
+ " <td>Female</td>\n",
1584
+ " <td>27.0</td>\n",
1585
+ " <td>Student</td>\n",
1586
+ " <td>5.0</td>\n",
1587
+ " <td>5.75</td>\n",
1588
+ " <td>5.0</td>\n",
1589
+ " <td>5-6 hours</td>\n",
1590
+ " <td>Unhealthy</td>\n",
1591
+ " <td>'Class 12'</td>\n",
1592
+ " <td>Yes</td>\n",
1593
+ " <td>7.0</td>\n",
1594
+ " <td>1.0</td>\n",
1595
+ " <td>Yes</td>\n",
1596
+ " <td>0</td>\n",
1597
+ " </tr>\n",
1598
+ " <tr>\n",
1599
+ " <th>27866</th>\n",
1600
+ " <td>Male</td>\n",
1601
+ " <td>27.0</td>\n",
1602
+ " <td>Student</td>\n",
1603
+ " <td>2.0</td>\n",
1604
+ " <td>9.40</td>\n",
1605
+ " <td>3.0</td>\n",
1606
+ " <td>Less than 5 hours</td>\n",
1607
+ " <td>Healthy</td>\n",
1608
+ " <td>MSc</td>\n",
1609
+ " <td>No</td>\n",
1610
+ " <td>0.0</td>\n",
1611
+ " <td>3.0</td>\n",
1612
+ " <td>Yes</td>\n",
1613
+ " <td>0</td>\n",
1614
+ " </tr>\n",
1615
+ " <tr>\n",
1616
+ " <th>27867</th>\n",
1617
+ " <td>Male</td>\n",
1618
+ " <td>31.0</td>\n",
1619
+ " <td>Student</td>\n",
1620
+ " <td>3.0</td>\n",
1621
+ " <td>6.61</td>\n",
1622
+ " <td>4.0</td>\n",
1623
+ " <td>5-6 hours</td>\n",
1624
+ " <td>Unhealthy</td>\n",
1625
+ " <td>MD</td>\n",
1626
+ " <td>No</td>\n",
1627
+ " <td>12.0</td>\n",
1628
+ " <td>2.0</td>\n",
1629
+ " <td>No</td>\n",
1630
+ " <td>0</td>\n",
1631
+ " </tr>\n",
1632
+ " <tr>\n",
1633
+ " <th>27868</th>\n",
1634
+ " <td>Female</td>\n",
1635
+ " <td>18.0</td>\n",
1636
+ " <td>Student</td>\n",
1637
+ " <td>5.0</td>\n",
1638
+ " <td>6.88</td>\n",
1639
+ " <td>2.0</td>\n",
1640
+ " <td>Less than 5 hours</td>\n",
1641
+ " <td>Healthy</td>\n",
1642
+ " <td>'Class 12'</td>\n",
1643
+ " <td>Yes</td>\n",
1644
+ " <td>10.0</td>\n",
1645
+ " <td>5.0</td>\n",
1646
+ " <td>No</td>\n",
1647
+ " <td>1</td>\n",
1648
+ " </tr>\n",
1649
+ " <tr>\n",
1650
+ " <th>27869</th>\n",
1651
+ " <td>Male</td>\n",
1652
+ " <td>27.0</td>\n",
1653
+ " <td>Student</td>\n",
1654
+ " <td>4.0</td>\n",
1655
+ " <td>9.24</td>\n",
1656
+ " <td>1.0</td>\n",
1657
+ " <td>Less than 5 hours</td>\n",
1658
+ " <td>Healthy</td>\n",
1659
+ " <td>BCA</td>\n",
1660
+ " <td>Yes</td>\n",
1661
+ " <td>2.0</td>\n",
1662
+ " <td>3.0</td>\n",
1663
+ " <td>Yes</td>\n",
1664
+ " <td>1</td>\n",
1665
+ " </tr>\n",
1666
+ " </tbody>\n",
1667
+ "</table>\n",
1668
+ "<p>27870 rows × 14 columns</p>\n",
1669
+ "</div>"
1670
+ ],
1671
+ "text/plain": [
1672
+ " Gender Age Profession Academic Pressure CGPA Study Satisfaction \\\n",
1673
+ "0 Male 33.0 Student 5.0 8.97 2.0 \n",
1674
+ "1 Female 24.0 Student 2.0 5.90 5.0 \n",
1675
+ "2 Male 31.0 Student 3.0 7.03 5.0 \n",
1676
+ "3 Female 28.0 Student 3.0 5.59 2.0 \n",
1677
+ "4 Female 25.0 Student 4.0 8.13 3.0 \n",
1678
+ "... ... ... ... ... ... ... \n",
1679
+ "27865 Female 27.0 Student 5.0 5.75 5.0 \n",
1680
+ "27866 Male 27.0 Student 2.0 9.40 3.0 \n",
1681
+ "27867 Male 31.0 Student 3.0 6.61 4.0 \n",
1682
+ "27868 Female 18.0 Student 5.0 6.88 2.0 \n",
1683
+ "27869 Male 27.0 Student 4.0 9.24 1.0 \n",
1684
+ "\n",
1685
+ " Sleep Duration Dietary Habits Degree \\\n",
1686
+ "0 5-6 hours Healthy B.Pharm \n",
1687
+ "1 5-6 hours Moderate BSc \n",
1688
+ "2 Less than 5 hours Healthy BA \n",
1689
+ "3 7-8 hours Moderate BCA \n",
1690
+ "4 5-6 hours Moderate M.Tech \n",
1691
+ "... ... ... ... \n",
1692
+ "27865 5-6 hours Unhealthy 'Class 12' \n",
1693
+ "27866 Less than 5 hours Healthy MSc \n",
1694
+ "27867 5-6 hours Unhealthy MD \n",
1695
+ "27868 Less than 5 hours Healthy 'Class 12' \n",
1696
+ "27869 Less than 5 hours Healthy BCA \n",
1697
+ "\n",
1698
+ " Have you ever had suicidal thoughts ? Work/Study Hours \\\n",
1699
+ "0 Yes 3.0 \n",
1700
+ "1 No 3.0 \n",
1701
+ "2 No 9.0 \n",
1702
+ "3 Yes 4.0 \n",
1703
+ "4 Yes 1.0 \n",
1704
+ "... ... ... \n",
1705
+ "27865 Yes 7.0 \n",
1706
+ "27866 No 0.0 \n",
1707
+ "27867 No 12.0 \n",
1708
+ "27868 Yes 10.0 \n",
1709
+ "27869 Yes 2.0 \n",
1710
+ "\n",
1711
+ " Financial Stress Family History of Mental Illness Depression \n",
1712
+ "0 1.0 No 1 \n",
1713
+ "1 2.0 Yes 0 \n",
1714
+ "2 1.0 Yes 0 \n",
1715
+ "3 5.0 Yes 1 \n",
1716
+ "4 1.0 No 0 \n",
1717
+ "... ... ... ... \n",
1718
+ "27865 1.0 Yes 0 \n",
1719
+ "27866 3.0 Yes 0 \n",
1720
+ "27867 2.0 No 0 \n",
1721
+ "27868 5.0 No 1 \n",
1722
+ "27869 3.0 Yes 1 \n",
1723
+ "\n",
1724
+ "[27870 rows x 14 columns]"
1725
+ ]
1726
+ },
1727
+ "execution_count": 178,
1728
+ "metadata": {},
1729
+ "output_type": "execute_result"
1730
+ }
1731
+ ],
1732
+ "source": [
1733
+ "df1"
1734
+ ]
1735
+ },
1736
+ {
1737
+ "cell_type": "code",
1738
+ "execution_count": 185,
1739
+ "id": "c55b07cc-488d-44a3-a782-a11d3e5031eb",
1740
+ "metadata": {},
1741
+ "outputs": [
1742
+ {
1743
+ "name": "stdout",
1744
+ "output_type": "stream",
1745
+ "text": [
1746
+ "Gender: ['Male' 'Female']\n",
1747
+ "Profession: ['Student']\n",
1748
+ "Sleep Duration: ['5-6 hours' 'Less than 5 hours' '7-8 hours' 'More than 8 hours' 'Others']\n",
1749
+ "Dietary Habits: ['Healthy' 'Moderate' 'Unhealthy' 'Others']\n",
1750
+ "Degree: ['B.Pharm' 'BSc' 'BA' 'BCA' 'M.Tech' 'PhD' 'Class 12' 'B.Ed' 'LLB' 'BE'\n",
1751
+ " 'M.Ed' 'MSc' 'BHM' 'M.Pharm' 'MCA' 'MA' 'B.Com' 'MD' 'MBA' 'MBBS' 'M.Com'\n",
1752
+ " 'B.Arch' 'LLM' 'B.Tech' 'BBA' 'ME' 'MHM' 'Others']\n",
1753
+ "Have you ever had suicidal thoughts ?: ['Yes' 'No']\n",
1754
+ "Financial Stress: ['1.0' '2.0' '5.0' '3.0' '4.0']\n",
1755
+ "Family History of Mental Illness: ['No' 'Yes']\n"
1756
+ ]
1757
+ }
1758
+ ],
1759
+ "source": [
1760
+ "for col in df1.select_dtypes(include=['object']).columns:\n",
1761
+ " print(f\"{col}: {df1[col].unique()}\")"
1762
+ ]
1763
+ },
1764
+ {
1765
+ "cell_type": "code",
1766
+ "execution_count": 189,
1767
+ "id": "aace3557-1ce9-4c9c-841b-5507bbd5ca4b",
1768
+ "metadata": {},
1769
+ "outputs": [
1770
+ {
1771
+ "data": {
1772
+ "text/plain": [
1773
+ "Gender 0\n",
1774
+ "Age 0\n",
1775
+ "Profession 0\n",
1776
+ "Academic Pressure 0\n",
1777
+ "CGPA 0\n",
1778
+ "Study Satisfaction 0\n",
1779
+ "Sleep Duration 0\n",
1780
+ "Dietary Habits 0\n",
1781
+ "Degree 0\n",
1782
+ "Have you ever had suicidal thoughts ? 0\n",
1783
+ "Work/Study Hours 0\n",
1784
+ "Financial Stress 0\n",
1785
+ "Family History of Mental Illness 0\n",
1786
+ "Depression 0\n",
1787
+ "dtype: int64"
1788
+ ]
1789
+ },
1790
+ "execution_count": 189,
1791
+ "metadata": {},
1792
+ "output_type": "execute_result"
1793
+ }
1794
+ ],
1795
+ "source": [
1796
+ "missing_values = df1.isnull().sum()\n",
1797
+ "missing_values"
1798
+ ]
1799
+ },
1800
+ {
1801
+ "cell_type": "code",
1802
+ "execution_count": 201,
1803
+ "id": "87e2b5d3-fa7d-4cc4-ac3c-8de0a1a4dd8b",
1804
+ "metadata": {},
1805
+ "outputs": [
1806
+ {
1807
+ "data": {
1808
+ "text/plain": [
1809
+ "0 3.0\n",
1810
+ "1 3.0\n",
1811
+ "2 9.0\n",
1812
+ "3 4.0\n",
1813
+ "4 1.0\n",
1814
+ " ... \n",
1815
+ "27865 7.0\n",
1816
+ "27866 0.0\n",
1817
+ "27867 12.0\n",
1818
+ "27868 10.0\n",
1819
+ "27869 2.0\n",
1820
+ "Name: Work/Study Hours, Length: 27870, dtype: float64"
1821
+ ]
1822
+ },
1823
+ "execution_count": 201,
1824
+ "metadata": {},
1825
+ "output_type": "execute_result"
1826
+ }
1827
+ ],
1828
+ "source": [
1829
+ "# outliers\n",
1830
+ "\n",
1831
+ "from scipy.stats.mstats import winsorize\n",
1832
+ "numerical_cols = [\"Age\", \"Academic Pressure\", \"CGPA\", \"Study Satisfaction\", \"Work/Study Hours\"]\n",
1833
+ "for col in numerical_cols:\n",
1834
+ " df1[col] = winsorize(df1[col], limits=[0.05, 0.05])\n",
1835
+ "\n",
1836
+ "df1[col]"
1837
+ ]
1838
+ },
1839
+ {
1840
+ "cell_type": "code",
1841
+ "execution_count": 209,
1842
+ "id": "e84385ca-f0b5-4dc7-8da7-ea89626330c1",
1843
+ "metadata": {},
1844
+ "outputs": [
1845
+ {
1846
+ "data": {
1847
+ "text/html": [
1848
+ "<div>\n",
1849
+ "<style scoped>\n",
1850
+ " .dataframe tbody tr th:only-of-type {\n",
1851
+ " vertical-align: middle;\n",
1852
+ " }\n",
1853
+ "\n",
1854
+ " .dataframe tbody tr th {\n",
1855
+ " vertical-align: top;\n",
1856
+ " }\n",
1857
+ "\n",
1858
+ " .dataframe thead th {\n",
1859
+ " text-align: right;\n",
1860
+ " }\n",
1861
+ "</style>\n",
1862
+ "<table border=\"1\" class=\"dataframe\">\n",
1863
+ " <thead>\n",
1864
+ " <tr style=\"text-align: right;\">\n",
1865
+ " <th></th>\n",
1866
+ " <th>Age</th>\n",
1867
+ " <th>Academic Pressure</th>\n",
1868
+ " <th>CGPA</th>\n",
1869
+ " <th>Study Satisfaction</th>\n",
1870
+ " <th>Work/Study Hours</th>\n",
1871
+ " </tr>\n",
1872
+ " </thead>\n",
1873
+ " <tbody>\n",
1874
+ " <tr>\n",
1875
+ " <th>0</th>\n",
1876
+ " <td>0.888889</td>\n",
1877
+ " <td>1.0</td>\n",
1878
+ " <td>0.456274</td>\n",
1879
+ " <td>-0.5</td>\n",
1880
+ " <td>-0.833333</td>\n",
1881
+ " </tr>\n",
1882
+ " <tr>\n",
1883
+ " <th>1</th>\n",
1884
+ " <td>-0.111111</td>\n",
1885
+ " <td>-0.5</td>\n",
1886
+ " <td>-0.711027</td>\n",
1887
+ " <td>1.0</td>\n",
1888
+ " <td>-0.833333</td>\n",
1889
+ " </tr>\n",
1890
+ " <tr>\n",
1891
+ " <th>2</th>\n",
1892
+ " <td>0.666667</td>\n",
1893
+ " <td>0.0</td>\n",
1894
+ " <td>-0.281369</td>\n",
1895
+ " <td>1.0</td>\n",
1896
+ " <td>0.166667</td>\n",
1897
+ " </tr>\n",
1898
+ " <tr>\n",
1899
+ " <th>3</th>\n",
1900
+ " <td>0.333333</td>\n",
1901
+ " <td>0.0</td>\n",
1902
+ " <td>-0.828897</td>\n",
1903
+ " <td>-0.5</td>\n",
1904
+ " <td>-0.666667</td>\n",
1905
+ " </tr>\n",
1906
+ " <tr>\n",
1907
+ " <th>4</th>\n",
1908
+ " <td>0.000000</td>\n",
1909
+ " <td>0.5</td>\n",
1910
+ " <td>0.136882</td>\n",
1911
+ " <td>0.0</td>\n",
1912
+ " <td>-1.166667</td>\n",
1913
+ " </tr>\n",
1914
+ " <tr>\n",
1915
+ " <th>...</th>\n",
1916
+ " <td>...</td>\n",
1917
+ " <td>...</td>\n",
1918
+ " <td>...</td>\n",
1919
+ " <td>...</td>\n",
1920
+ " <td>...</td>\n",
1921
+ " </tr>\n",
1922
+ " <tr>\n",
1923
+ " <th>27865</th>\n",
1924
+ " <td>0.222222</td>\n",
1925
+ " <td>1.0</td>\n",
1926
+ " <td>-0.768061</td>\n",
1927
+ " <td>1.0</td>\n",
1928
+ " <td>-0.166667</td>\n",
1929
+ " </tr>\n",
1930
+ " <tr>\n",
1931
+ " <th>27866</th>\n",
1932
+ " <td>0.222222</td>\n",
1933
+ " <td>-0.5</td>\n",
1934
+ " <td>0.619772</td>\n",
1935
+ " <td>0.0</td>\n",
1936
+ " <td>-1.333333</td>\n",
1937
+ " </tr>\n",
1938
+ " <tr>\n",
1939
+ " <th>27867</th>\n",
1940
+ " <td>0.666667</td>\n",
1941
+ " <td>0.0</td>\n",
1942
+ " <td>-0.441065</td>\n",
1943
+ " <td>0.5</td>\n",
1944
+ " <td>0.666667</td>\n",
1945
+ " </tr>\n",
1946
+ " <tr>\n",
1947
+ " <th>27868</th>\n",
1948
+ " <td>-0.777778</td>\n",
1949
+ " <td>1.0</td>\n",
1950
+ " <td>-0.338403</td>\n",
1951
+ " <td>-0.5</td>\n",
1952
+ " <td>0.333333</td>\n",
1953
+ " </tr>\n",
1954
+ " <tr>\n",
1955
+ " <th>27869</th>\n",
1956
+ " <td>0.222222</td>\n",
1957
+ " <td>0.5</td>\n",
1958
+ " <td>0.558935</td>\n",
1959
+ " <td>-1.0</td>\n",
1960
+ " <td>-1.000000</td>\n",
1961
+ " </tr>\n",
1962
+ " </tbody>\n",
1963
+ "</table>\n",
1964
+ "<p>27870 rows × 5 columns</p>\n",
1965
+ "</div>"
1966
+ ],
1967
+ "text/plain": [
1968
+ " Age Academic Pressure CGPA Study Satisfaction \\\n",
1969
+ "0 0.888889 1.0 0.456274 -0.5 \n",
1970
+ "1 -0.111111 -0.5 -0.711027 1.0 \n",
1971
+ "2 0.666667 0.0 -0.281369 1.0 \n",
1972
+ "3 0.333333 0.0 -0.828897 -0.5 \n",
1973
+ "4 0.000000 0.5 0.136882 0.0 \n",
1974
+ "... ... ... ... ... \n",
1975
+ "27865 0.222222 1.0 -0.768061 1.0 \n",
1976
+ "27866 0.222222 -0.5 0.619772 0.0 \n",
1977
+ "27867 0.666667 0.0 -0.441065 0.5 \n",
1978
+ "27868 -0.777778 1.0 -0.338403 -0.5 \n",
1979
+ "27869 0.222222 0.5 0.558935 -1.0 \n",
1980
+ "\n",
1981
+ " Work/Study Hours \n",
1982
+ "0 -0.833333 \n",
1983
+ "1 -0.833333 \n",
1984
+ "2 0.166667 \n",
1985
+ "3 -0.666667 \n",
1986
+ "4 -1.166667 \n",
1987
+ "... ... \n",
1988
+ "27865 -0.166667 \n",
1989
+ "27866 -1.333333 \n",
1990
+ "27867 0.666667 \n",
1991
+ "27868 0.333333 \n",
1992
+ "27869 -1.000000 \n",
1993
+ "\n",
1994
+ "[27870 rows x 5 columns]"
1995
+ ]
1996
+ },
1997
+ "execution_count": 209,
1998
+ "metadata": {},
1999
+ "output_type": "execute_result"
2000
+ }
2001
+ ],
2002
+ "source": [
2003
+ "from sklearn.preprocessing import RobustScaler\n",
2004
+ "scaler = RobustScaler()\n",
2005
+ "df1[numerical_cols] = scaler.fit_transform(df1[numerical_cols])\n",
2006
+ "df1[numerical_cols]"
2007
+ ]
2008
+ },
2009
+ {
2010
+ "cell_type": "code",
2011
+ "execution_count": 211,
2012
+ "id": "1c0d1179-8363-4f40-88d7-54f63f898bc0",
2013
+ "metadata": {},
2014
+ "outputs": [
2015
+ {
2016
+ "data": {
2017
+ "text/html": [
2018
+ "<div>\n",
2019
+ "<style scoped>\n",
2020
+ " .dataframe tbody tr th:only-of-type {\n",
2021
+ " vertical-align: middle;\n",
2022
+ " }\n",
2023
+ "\n",
2024
+ " .dataframe tbody tr th {\n",
2025
+ " vertical-align: top;\n",
2026
+ " }\n",
2027
+ "\n",
2028
+ " .dataframe thead th {\n",
2029
+ " text-align: right;\n",
2030
+ " }\n",
2031
+ "</style>\n",
2032
+ "<table border=\"1\" class=\"dataframe\">\n",
2033
+ " <thead>\n",
2034
+ " <tr style=\"text-align: right;\">\n",
2035
+ " <th></th>\n",
2036
+ " <th>Age</th>\n",
2037
+ " <th>Academic Pressure</th>\n",
2038
+ " <th>CGPA</th>\n",
2039
+ " <th>Study Satisfaction</th>\n",
2040
+ " <th>Work/Study Hours</th>\n",
2041
+ " </tr>\n",
2042
+ " </thead>\n",
2043
+ " <tbody>\n",
2044
+ " <tr>\n",
2045
+ " <th>count</th>\n",
2046
+ " <td>27870.000000</td>\n",
2047
+ " <td>27870.000000</td>\n",
2048
+ " <td>27870.000000</td>\n",
2049
+ " <td>27870.000000</td>\n",
2050
+ " <td>27870.000000</td>\n",
2051
+ " </tr>\n",
2052
+ " <tr>\n",
2053
+ " <th>mean</th>\n",
2054
+ " <td>0.089910</td>\n",
2055
+ " <td>0.070470</td>\n",
2056
+ " <td>-0.040597</td>\n",
2057
+ " <td>-0.028041</td>\n",
2058
+ " <td>-0.140342</td>\n",
2059
+ " </tr>\n",
2060
+ " <tr>\n",
2061
+ " <th>std</th>\n",
2062
+ " <td>0.541198</td>\n",
2063
+ " <td>0.690511</td>\n",
2064
+ " <td>0.550120</td>\n",
2065
+ " <td>0.680283</td>\n",
2066
+ " <td>0.617863</td>\n",
2067
+ " </tr>\n",
2068
+ " <tr>\n",
2069
+ " <th>min</th>\n",
2070
+ " <td>-0.777778</td>\n",
2071
+ " <td>-1.000000</td>\n",
2072
+ " <td>-0.908745</td>\n",
2073
+ " <td>-1.000000</td>\n",
2074
+ " <td>-1.333333</td>\n",
2075
+ " </tr>\n",
2076
+ " <tr>\n",
2077
+ " <th>25%</th>\n",
2078
+ " <td>-0.444444</td>\n",
2079
+ " <td>-0.500000</td>\n",
2080
+ " <td>-0.562738</td>\n",
2081
+ " <td>-0.500000</td>\n",
2082
+ " <td>-0.666667</td>\n",
2083
+ " </tr>\n",
2084
+ " <tr>\n",
2085
+ " <th>50%</th>\n",
2086
+ " <td>0.000000</td>\n",
2087
+ " <td>0.000000</td>\n",
2088
+ " <td>0.000000</td>\n",
2089
+ " <td>0.000000</td>\n",
2090
+ " <td>0.000000</td>\n",
2091
+ " </tr>\n",
2092
+ " <tr>\n",
2093
+ " <th>75%</th>\n",
2094
+ " <td>0.555556</td>\n",
2095
+ " <td>0.500000</td>\n",
2096
+ " <td>0.437262</td>\n",
2097
+ " <td>0.500000</td>\n",
2098
+ " <td>0.333333</td>\n",
2099
+ " </tr>\n",
2100
+ " <tr>\n",
2101
+ " <th>max</th>\n",
2102
+ " <td>1.000000</td>\n",
2103
+ " <td>1.000000</td>\n",
2104
+ " <td>0.806084</td>\n",
2105
+ " <td>1.000000</td>\n",
2106
+ " <td>0.666667</td>\n",
2107
+ " </tr>\n",
2108
+ " </tbody>\n",
2109
+ "</table>\n",
2110
+ "</div>"
2111
+ ],
2112
+ "text/plain": [
2113
+ " Age Academic Pressure CGPA Study Satisfaction \\\n",
2114
+ "count 27870.000000 27870.000000 27870.000000 27870.000000 \n",
2115
+ "mean 0.089910 0.070470 -0.040597 -0.028041 \n",
2116
+ "std 0.541198 0.690511 0.550120 0.680283 \n",
2117
+ "min -0.777778 -1.000000 -0.908745 -1.000000 \n",
2118
+ "25% -0.444444 -0.500000 -0.562738 -0.500000 \n",
2119
+ "50% 0.000000 0.000000 0.000000 0.000000 \n",
2120
+ "75% 0.555556 0.500000 0.437262 0.500000 \n",
2121
+ "max 1.000000 1.000000 0.806084 1.000000 \n",
2122
+ "\n",
2123
+ " Work/Study Hours \n",
2124
+ "count 27870.000000 \n",
2125
+ "mean -0.140342 \n",
2126
+ "std 0.617863 \n",
2127
+ "min -1.333333 \n",
2128
+ "25% -0.666667 \n",
2129
+ "50% 0.000000 \n",
2130
+ "75% 0.333333 \n",
2131
+ "max 0.666667 "
2132
+ ]
2133
+ },
2134
+ "execution_count": 211,
2135
+ "metadata": {},
2136
+ "output_type": "execute_result"
2137
+ }
2138
+ ],
2139
+ "source": [
2140
+ "df1[numerical_cols].describe()"
2141
+ ]
2142
+ },
2143
+ {
2144
+ "cell_type": "code",
2145
+ "execution_count": null,
2146
+ "id": "f10ac51a-c19d-4a75-860f-a6e048351b45",
2147
+ "metadata": {},
2148
+ "outputs": [],
2149
+ "source": []
2150
+ }
2151
+ ],
2152
+ "metadata": {
2153
+ "kernelspec": {
2154
+ "display_name": "Python 3 (ipykernel)",
2155
+ "language": "python",
2156
+ "name": "python3"
2157
+ },
2158
+ "language_info": {
2159
+ "codemirror_mode": {
2160
+ "name": "ipython",
2161
+ "version": 3
2162
+ },
2163
+ "file_extension": ".py",
2164
+ "mimetype": "text/x-python",
2165
+ "name": "python",
2166
+ "nbconvert_exporter": "python",
2167
+ "pygments_lexer": "ipython3",
2168
+ "version": "3.12.4"
2169
+ }
2170
+ },
2171
+ "nbformat": 4,
2172
+ "nbformat_minor": 5
2173
+ }
app.py ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pandas as pd
3
+ import matplotlib.pyplot as plt
4
+ from sklearn.model_selection import train_test_split
5
+ from sklearn.preprocessing import LabelEncoder
6
+ from sklearn.ensemble import RandomForestClassifier
7
+ from sklearn.metrics import accuracy_score, classification_report
8
+
9
+ def load_data(uploaded_file):
10
+ df = pd.read_csv(uploaded_file)
11
+ df.drop(columns=['id', 'City', 'Profession', 'Degree'], inplace=True, errors='ignore')
12
+ return df
13
+
14
+ def preprocess_data(df):
15
+ label_encoders = {}
16
+ categorical_cols = ['Gender', 'Sleep Duration', 'Dietary Habits',
17
+ 'Have you ever had suicidal thoughts ?', 'Financial Stress',
18
+ 'Family History of Mental Illness']
19
+ for col in categorical_cols:
20
+ if col in df.columns:
21
+ le = LabelEncoder()
22
+ df[col] = le.fit_transform(df[col])
23
+ label_encoders[col] = le
24
+ return df, label_encoders
25
+
26
+ def train_model(X_train, y_train):
27
+ model = RandomForestClassifier(n_estimators=100, random_state=42)
28
+ model.fit(X_train, y_train)
29
+ return model
30
+
31
+ def main():
32
+ st.title("Student Depression Prediction")
33
+ st.write("Upload your dataset to train and evaluate the Random Forest model.")
34
+
35
+ uploaded_file = st.file_uploader("Upload CSV", type=["csv"])
36
+
37
+ if uploaded_file:
38
+ df = load_data(uploaded_file)
39
+ df, _ = preprocess_data(df)
40
+
41
+ X = df.drop(columns=['Depression'])
42
+ y = df['Depression']
43
+ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42, stratify=y)
44
+
45
+ model = train_model(X_train, y_train)
46
+ y_pred = model.predict(X_test)
47
+
48
+ accuracy = accuracy_score(y_test, y_pred)
49
+ classification_rep = classification_report(y_test, y_pred)
50
+
51
+ feature_importances = model.feature_importances_
52
+ feature_importance_df = pd.DataFrame({
53
+ 'Feature': X.columns,
54
+ 'Importance': feature_importances
55
+ }).sort_values(by='Importance', ascending=False)
56
+
57
+ st.subheader("Model Evaluation")
58
+ st.write(f"**Accuracy:** {accuracy:.2f}")
59
+ st.text("Classification Report:")
60
+ st.text(classification_rep)
61
+
62
+ st.subheader("Feature Importance")
63
+ fig, ax = plt.subplots()
64
+ ax.barh(feature_importance_df['Feature'], feature_importance_df['Importance'], color='skyblue')
65
+ ax.set_xlabel("Importance")
66
+ ax.set_ylabel("Feature")
67
+ ax.set_title("Feature Importance in Random Forest")
68
+ st.pyplot(fig)
student_depression_dataset.csv ADDED
The diff for this file is too large to render. See raw diff