stressUp / stress_model /final /config.json
YureiYuri's picture
stressUp model
631fd05 verified
raw
history blame contribute delete
674 Bytes
{
"input_dim": 20,
"num_classes": 3,
"dropout": 0.3,
"feature_cols": [
"anxiety_level",
"self_esteem",
"mental_health_history",
"depression",
"headache",
"blood_pressure",
"sleep_quality",
"breathing_problem",
"noise_level",
"living_conditions",
"safety",
"basic_needs",
"academic_performance",
"study_load",
"teacher_student_relationship",
"future_career_concerns",
"social_support",
"peer_pressure",
"extracurricular_activities",
"bullying"
],
"id2label": {
"0": "low",
"1": "medium",
"2": "high"
},
"label2id": {
"low": 0,
"medium": 1,
"high": 2
}
}