depression_example / config.json
silvaKenpachi's picture
Create config.json
e23ea6e verified
{
"sklearn": {
"columns": [
"Name",
"Gender",
"City",
"Working Professional or Student",
"Profession",
"Sleep Duration",
"Dietary Habits",
"Degree",
"Have you ever had suicidal thoughts ?",
"Family History of Mental Illness",
"Academic Pressure_na",
"Work Pressure_na",
"CGPA_na",
"Study Satisfaction_na",
"Job Satisfaction_na",
"Financial Stress_na",
"Age",
"Academic Pressure",
"Work Pressure",
"CGPA",
"Study Satisfaction",
"Job Satisfaction",
"Work/Study Hours",
"Financial Stress"
],
"environment": [
"scikit-learn=1.3.2"
],
"example_input": {
"Academic Pressure": [
-0.04606674611568451,
-0.04606674611568451,
-0.04606674611568451
],
"Academic Pressure_na": [
2,
2,
2
],
"Age": [
1.2607895135879517,
-0.19406381249427795,
1.584090232849121
],
"CGPA": [
0.03309895098209381,
0.03309895098209381,
0.03309895098209381
],
"CGPA_na": [
2,
2,
2
],
"City": [
98,
28,
27
],
"Degree": [
77,
90,
77
],
"Dietary Habits": [
16,
21,
21
],
"Family History of Mental Illness": [
1,
2,
2
],
"Financial Stress": [
1.4229286909103394,
1.4229286909103394,
0.008482186123728752
],
"Financial Stress_na": [
1,
1,
1
],
"Gender": [
2,
2,
2
],
"Have you ever had suicidal thoughts ?": [
1,
1,
2
],
"Job Satisfaction": [
0.8020281791687012,
1.5897727012634277,
-1.5612051486968994
],
"Job Satisfaction_na": [
1,
1,
1
],
"Name": [
79,
269,
90
],
"Profession": [
46,
15,
56
],
"Sleep Duration": [
30,
20,
30
],
"Study Satisfaction": [
0.019511962309479713,
0.019511962309479713,
0.019511962309479713
],
"Study Satisfaction_na": [
2,
2,
2
],
"Work Pressure": [
1.588835597038269,
-0.7938991189002991,
-0.7938991189002991
],
"Work Pressure_na": [
1,
1,
1
],
"Work/Study Hours": [
1.2296171188354492,
1.4891586303710938,
-1.1062564849853516
],
"Working Professional or Student": [
2,
2,
2
]
},
"model": {
"file": "model.pkl"
},
"model_format": "pickle",
"task": "tabular-classification"
}
}