Update app.py
Browse files
app.py
CHANGED
@@ -39,12 +39,12 @@ def titanic(ticket_class, sex, port, fare, age, sibsp, parch):
|
|
39 |
input_list.append(int(sibsp))
|
40 |
input_list.append(int(parch))
|
41 |
data = {
|
42 |
-
"pclass": CLASS_TO_VALUE[ticket_class],
|
43 |
-
"sex": sex,
|
44 |
-
"embarked": PORT_TO_VALUE[port],
|
45 |
-
"age": age,
|
46 |
-
"sibsp": int(sibsp),
|
47 |
-
"parch": int(parch)
|
48 |
}
|
49 |
df = pd.DataFrame(data)
|
50 |
# 'res' is a list of predictions returned as the label.
|
|
|
39 |
input_list.append(int(sibsp))
|
40 |
input_list.append(int(parch))
|
41 |
data = {
|
42 |
+
"pclass": [CLASS_TO_VALUE[ticket_class]],
|
43 |
+
"sex": [sex],
|
44 |
+
"embarked": [PORT_TO_VALUE[port]],
|
45 |
+
"age": [age],
|
46 |
+
"sibsp": [int(sibsp)],
|
47 |
+
"parch": [int(parch)],
|
48 |
}
|
49 |
df = pd.DataFrame(data)
|
50 |
# 'res' is a list of predictions returned as the label.
|