pantdipendra
commited on
final test
Browse files
app.py
CHANGED
@@ -36,22 +36,22 @@ class ModelPredictor:
|
|
36 |
# - "1" = first item in list
|
37 |
# - "2" = second item
|
38 |
self.prediction_map = {
|
39 |
-
"YOWRCONC": {
|
40 |
-
"YOSEEDOC": {
|
41 |
-
"YO_MDEA5": {
|
42 |
-
"YOWRLSIN": {
|
43 |
-
"YODPPROB": {
|
44 |
-
"YOWRPROB": {
|
45 |
-
"YODPR2WK": {
|
46 |
-
"YOWRDEPR": {
|
47 |
-
"YODPDISC": {
|
48 |
-
"YOLOSEV": {
|
49 |
-
"YOWRDCSN": {
|
50 |
-
"YODSMMDE": {
|
51 |
-
"YO_MDEA3": {
|
52 |
-
"YODPLSIN": {
|
53 |
-
"YOWRELES": {
|
54 |
-
"YOPB2WK": {
|
55 |
}
|
56 |
|
57 |
def load_models(self):
|
|
|
36 |
# - "1" = first item in list
|
37 |
# - "2" = second item
|
38 |
self.prediction_map = {
|
39 |
+
"YOWRCONC": {2: "Did NOT have difficulty concentrating", 1: "Had difficulty concentrating"},
|
40 |
+
"YOSEEDOC": {2: "Did NOT feel the need to see a doctor", 1: "Felt the need to see a doctor"},
|
41 |
+
"YO_MDEA5": {2: "No restlessness/lethargy noticed", 1: "Others noticed restlessness/lethargy"},
|
42 |
+
"YOWRLSIN": {2: "Did NOT feel bored/lose interest", 1: "Felt bored/lost interest"},
|
43 |
+
"YODPPROB": {2: "No other problems for 2+ weeks", 1: "Had other problems for 2+ weeks"},
|
44 |
+
"YOWRPROB": {2: "No 'worst time ever' feeling", 1: "Had 'worst time ever' feeling"},
|
45 |
+
"YODPR2WK": {2: "No depressed feelings for 2+ wks", 1: "Had depressed feelings for 2+ wks"},
|
46 |
+
"YOWRDEPR": {2: "Did NOT feel sad/depressed daily", 1: "Felt sad/depressed mostly everyday"},
|
47 |
+
"YODPDISC": {2: "Overall mood not sad/depressed", 1: "Overall mood was sad/depressed"},
|
48 |
+
"YOLOSEV": {2: "Did NOT lose interest in things", 1: "Lost interest in enjoyable things"},
|
49 |
+
"YOWRDCSN": {2: "Was able to make decisions", 1: "Was unable to make decisions"},
|
50 |
+
"YODSMMDE": {2: "No 2+ wks depression symptoms", 1: "Had 2+ wks depression symptoms"},
|
51 |
+
"YO_MDEA3": {2: "No appetite/weight changes", 1: "Had changes in appetite/weight"},
|
52 |
+
"YODPLSIN": {2: "Never lost interest/felt bored", 1: "Lost interest/felt bored"},
|
53 |
+
"YOWRELES": {2: "Did NOT eat less than usual", 1: "Ate less than usual"},
|
54 |
+
"YOPB2WK": {2: "No uneasy feelings 2+ weeks", 1: "Uneasy feelings 2+ weeks"}
|
55 |
}
|
56 |
|
57 |
def load_models(self):
|